数据结构 uva-540-Team Queue

 

题目链接：

http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&category=&problem=481&mosmsg=Submission+received+with+ID+11144937

 

题目意思：

    给你n个组，以及每组的成员，让你建一个队列，开始队列为空。

有两种操作：

1、ENQUEUE x  ： 如果该队列中没有x的组友，就将x插入到队列的最后。如果有就插到该组的最后一个元素后面。

2、DEQUEUE     ： 出队操作，和普通队列一样执行操作。

 

解题思路：

问题的关键是怎样快速定位的插入的位置。分析知，如果该队列有组友，只会插到该组的最后一个位置即可，如果没有就插到队列的最后位置。

于是就可以想到用last[i]保存每组的最后一个元素的iterator，用team[i]表示i所属的队列。

 

代码：

#include<iostream>#include<cmath>#include<cstdio>#include<cstdlib>#include<string>#include<cstring>#include<algorithm>#include<vector>#include<map>#include<stack>#include<queue>#include<list>#define eps 1e-6#define INF (1<<20)#define PI acos(-1.0)using namespace std;int team[1000000];int main(){    int n,ca(0);    int temp,cur;    char order[20];    while(scanf("%d",&n)!=EOF&&n)    {        for(int i=1;i<=n;i++)        {            scanf("%d",&temp);            for(int j=1;j<=temp;j++)            {                scanf("%d",&cur);                team[cur]=i;        //元素所属的队列            }        }        list<int>mylist,templist;   //templist 仅仅是作为一个常量标记结尾        list<int>::iterator last[1010];        for(int i=1;i<=n;i++)            last[i]=templist.end();  //开始每组在队列中的元素为零        printf("Scenario #%d\n",++ca);        while(scanf("%s",order)&&strcmp(order,"STOP"))        {            if(strcmp(order,"ENQUEUE")==0)            {                scanf("%d",&cur);                if(last[team[cur]]==templist.end())  //如果队列中还没有该组成员                    last[team[cur]]=mylist.insert(mylist.end(),cur); //将该元素插到队列的最后                else                {                    last[team[cur]]++;     //如果队列中又该组的成员                    last[team[cur]]=mylist.insert(last[team[cur]],cur); //插到该组的最后一个元素后面                }            }            else            {                cur=mylist.front();                printf("%d\n",cur);                list<int>::iterator tt;                tt=mylist.begin();                                if(tt==last[team[cur]])   //注意判断一下出队的元素是不是最后一个元素，否则的话就要改变last的值                    last[team[cur]]=templist.end();                                               mylist.pop_front();              }        }        putchar('\n');    }}