UVA-540 Team Queue

2016-08-22

UVA - 540 Team Queue

题目大意：t 个队伍排队，若队伍中已经有自己队伍的人，则排到这个队伍的最后面，若没有，排到整个队列的后面。

解题思路：模拟，映射把每个人和队伍号对应起来，定义两个队列，一个存放排队的团队的编号，另一个记录某个队伍已排人的编号。

注意：要用STL，不然容易TLE。

#include <cstdio>
#include <iostream>
#include <map>
#include <queue>
using namespace std;

int main() {
	int t, kase = 0;
	while ( ~scanf("%d", &t) && t ) {
		cout << "Scenario #" << ++kase << endl;
		map<int, int> team; //team[x]表示编号为x的人所在的团队的编号。
		for (int i = 0; i < t; i++) {
			int n, x;
			scanf("%d", &n);
			while ( n-- ) {
				scanf("%d", &x);
				team[x] = i;
			}
		}
		queue<int> q, q2[1010];//q存放的是团队的编号，q2[i]是团队i成员的队列
		while (1) {
			int x;
			char str[100];
			scanf("%s", str);
			if ( str[0] == 'S' )
				break;
			else if ( str[0] == 'E' ) {
				scanf("%d", &x);
				int t = team[x];
				if ( q2[t].empty() )
					q.push(t);//团队t进入队列
				q2[t].push(x);
			}
			else if ( str[0] == 'D' ) {
				int t = q.front();
				cout << q2[t].front() << endl;
				q2[t].pop();
				if ( q2[t].empty() )
					q.pop();
			}
		}
		cout << endl;
	}
	return 0;
}