公司食堂-美团笔试
链接:https://www.nowcoder.com/questionTerminal/601815bea5544f389bcd20fb5ebca6a8
来源:牛客网
小美和小团所在公司的食堂有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
输出
2 1 1 3 4 4
思路:5*1e5二重for超时。考虑两个优先队列,一个存放0人桌子,另一个存放一人桌子。当选择选择0人桌时,0人桌出队,入队一人桌。当选则1人桌时,1人桌出队。
#include<iostream> #include<cstring> #include<queue> using namespace std; int main(){ int n; scanf("%d", &n); while(n--){ priority_queue<int, vector<int>, greater<int> > q1; priority_queue<int, vector<int>, greater<int> > q2; int N, M; string s, member; cin>>N; cin>>s; cin>>M; cin>>member; //scanf("%d", &N); //scanf("%s", &s); ///scanf("%d", &M); //scanf("%s", &member); for(int i=0; i<N; i++){ if(s[i]=='0') q1.push(i+1); else if(s[i]=='1') q2.push(i+1); } for(int i =0; i<member.length(); i++){ if(member[i]=='M'){ if(!q2.empty()){ int temp = q2.top(); printf("%d\n", temp); q2.pop(); } else{ int temp = q1.top(); printf("%d\n", temp); q1.pop(); q2.push(temp); } } else{ if(!q1.empty()){ int temp = q1.top(); printf("%d\n", temp); q1.pop(); q2.push(temp); } else{ int temp = q2.top(); printf("%d\n", temp); q2.pop(); } } } //cout<<"---"<<endl; } return 0; }