题意:有n只team,每只team有一些元素。然后这些元素按照队列的规律排队,如果一个元素在队伍中能够找到属于相同team的元素,则排在这些属于相同team的元素的最后方,否则作为队伍的第一个元素排在队列的最后。值得注意的是,题目输入数据都是在队伍列表中。
思路:对于如此大规模的数据量,在队列中搜索队友是很耗时的。最好的办法是用映射,鉴于本题数据是数字,所以可以用一个数组建立映射,下标为元素的数字,值为对应的队伍编号。由此可以很快的搜索到队友。另外再利用一个数组判断队列中是否存在队友。
//540 Team Queue
#include <iostream>
#include <queue>
using namespace std;
const int maxn = 1010;
int team_num[1000000];
int main()
{
//freopen("data.txt", "r", stdin);
int t;
int cases = 1;
while (scanf("%d", &t) && t)
{
for(int i = 1; i <= t; i++)
{
int n;
scanf("%d", &n);
for(int j = 0; j < n; j++)
{
int team;
scanf("%d", &team);
team_num[team] = i;//把元素映射到队伍编号上
}
}
printf("Scenario #%d\n", cases++);
bool in_queue[maxn] = {0};
char str[30];
queue<int> team;
queue<int> team_q[maxn];
while (scanf("%s", str))
{
if(str[0] == 'S')
break;
if(str[0] == 'E')
{
int tmp;
scanf("%d", &tmp);
int te = team_num[tmp];//映射得到队伍编号
if(!in_queue[te])//如果还没有队友在队列
{
team.push(te);//让元素的队伍编号入列
team_q[te].push(tmp);//让元素入列
in_queue[te] = true;//标记该队已入列
}
else
team_q[te].push(tmp);//放在队友后面
}
if(str[0] == 'D')
{
int te = team.front();
printf("%d\n", team_q[te].front());
team_q[te].pop();
if(team_q[te].empty())
{
team.pop();
in_queue[te] = false;
}
}
}
printf("\n");
}
return 0;
}