美团笔试题——公司食堂

122 篇文章 6 订阅

题目描述

小美和小团所在公司的食堂有N张餐桌,从左到右摆成一排,每张餐桌有2张餐椅供至多2人用餐,公司职员排队进入食堂用餐。小美发现职员用餐的一个规律并告诉小团:当男职员进入食堂时,他会优先选择已经坐有1人的餐桌用餐,只有当每张餐桌要么空着要么坐满2人时,他才会考虑空着的餐桌;

当女职员进入食堂时,她会优先选择未坐人的餐桌用餐,只有当每张餐桌都坐有至少1人时,她才会考虑已经坐有1人的餐桌;

无论男女,当有多张餐桌供职员选择时,他会选择最靠左的餐桌用餐。现在食堂内已有若干人在用餐,另外M个人正排队进入食堂,小团会根据小美告诉他的规律预测排队的每个人分别会坐哪张餐桌。

输入描述:

第一行输入一个整数T(1<=T<=10),表示数据组数。
每组数据占四行,第一行输入一个整数N(1<=N<=500000);
第二行输入一个长度为N且仅包含数字0、1、2的字符串,第i个数字表示左起第i张餐桌已坐有的用餐人数;
第三行输入一个整数M(1<=M<=2N且保证排队的每个人进入食堂时都有可供选择的餐桌);
第四行输入一个长度为M且仅包含字母M、F的字符串,若第i个字母为M,则排在第i的人为男性,否则其为女性。

输出描述:

每组数据输出占M行,第i行输出一个整数j(1<=j<=N),表示排在第i的人将选择左起第j张餐桌用餐。

输入例子1:

1
5
01102
6
MFMMFF

输出例子1:

2
1
1
3
4
4

题解
使用两个队列分别来维护座位人数为0的位置信息以及座位人数为1的位置信息
其中用普通队列来维护座位人数为0的位置信息
用优先队列来维护座位人数为1的位置信息(从小到大排序)
如果当前为女性,则从队列一中弹出一个位置信息进行记录,并加入到队列二中
否则,从队列二中弹出一个位置信息。

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
int a[N],b[N];
int n, m;
char s1[N],s2[N];
int main()
{
    int t;
    cin >> t;
    while(t--){
        queue<int> q1;
        priority_queue<int,vector<int>,greater<int>> q2;
        scanf("%d", &n);
        scanf("%s",s1);
        for(int i = 0; i < n; i++){
            if(s1[i] == '0')q1.push(i + 1);
            if(s1[i] == '1')q2.push(i + 1);
        }
        scanf("%d", &m);
        scanf("%s",s2);
        for(int i = 0; i < m; i++){
            if(s2[i] == 'M'){
                if(q2.empty()){
                    b[i] = q1.front();
                    q1.pop();
                    q2.push(b[i]);
 
                }
                else {
                    b[i] = q2.top();
                    q2.pop();
                }
            }
            else{
                if(q1.empty()){
                    b[i] = q2.top();
                    q2.pop();
                }
                else{
                    b[i] = q1.front();
                    q1.pop();
                    q2.push(b[i]);
                }
 
            }
        }
        for(int i = 0; i < m; i++)
            printf("%d\n",b[i]);
    }
    return 0;
}
  • 30
    点赞
  • 1
    评论
  • 0
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

打赏
文章很值,打赏犒劳作者一下
相关推荐
©️2020 CSDN 皮肤主题: 数字20 设计师:CSDN官方博客 返回首页

打赏

逐梦er

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值