2021-3-23-美团-优先队列

Posted 2022-12-07 茴薏

题目描述：
小美和小团所在公司的食堂有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

#include <iostream>
#include <vector>
#include <string>

using namespace std;

int main()

	int T  = 0;
	cin >> T;

	for (int k = 0; k < T; k++)
	
		int N , M ;
		cin >> N;
		string desk, que;
		cin >> M;
		cin >> desk;
		cin >> que;

		int f_zero = -1, f_one = -1;

		for (int i = 0; i < N; i++)
		
			if (f_zero == -1 && f_one == -1)
			
				break;
			
			else if (f_zero == -1 && desk[i] == '0')
			
				f_zero = i;
			
			else if (f_one == -1 && desk[i] == '1')
			
				f_one = i;
			
		

		for (int i = 0; i < M; i++)
		
			if (que[i] == 'M')
			
				//当男职员进入食堂时， 会优先选择已经做有1人的餐桌用餐
				if (f_one < N)
				
					cout << f_one + 1 << endl;
					desk[f_one] = '2';
					while (f_one < N && desk[f_one] != '1')
					
						f_one++;
					
				
				else
				
					//只有当每张餐桌要么空着要么坐满2人， 他才会考虑空着的餐桌
					cout << f_zero + 1 << endl;
					desk[f_zero] = '1';
					if (f_zero < f_one)
					
						f_one = f_zero;
					
					while (f_zero < N && desk[f_zero] != '0')
					
						f_zero++;
					
				
			
			else
			
				//当女职员进入食堂时，会优先选择未做人的餐桌用餐
				if (f_zero < N)
				
					cout << f_zero + 1 << endl;
					desk[f_zero] = '1';
					while (f_zero < N && desk[f_zero] != '0')
					
						f_zero++;
					
				
				else
				
					//只有当每张餐桌都只做1人时，她才考虑已经坐满1人的餐桌
					cout << f_one + 1 << endl;
					desk[f_one] = '2';
					while (f_one < N && desk[f_one] != '1')
					
						f_one++;