数据结构 queue

问题描述

t 个团队在餐厅前准备排队。他们的排队规则是：
初始队伍为空。一个人要排进队伍前，先搜索队伍中是否有他的队友。如果
有，这名成员就直接站在最后一个队友的后面，如果没有，那么这名成员只能排
在整个队伍的最后面。排队中途，队首的人可能被要求离开队伍。
依照上述排队规则，给出一些操作，操作有以下两种：
(1) IN x ， – 编号为 x 的成员进入队伍；
(2) OUT ， – 队首成员离开队伍。
操作结束后按顺序输出所有离开成员的编号。

★数据输入
输入第一行为一个正整数 t,代表团队的数量(1<=t<=1000)。
接下来 t 行，每行第一个整数为该团队的人数 n(1<=n<=1000)，接着 n 个整
数代表 n 个成员的编号 id， id 唯一且 1<=id<=1000000。
接下来 q 行操作(1<=q<=200000)，注意可能出现已经在队列中的人重复入队
的情况，忽略这样的操作。

★数据输出
输出第一行为整数 m，离开队伍的成员的数量。
接下来 m 行按序输出离开的成员的 id。

输入示例	输出示例
2 3 101 102 105 3 103 104 106 12 IN 101 IN 103 IN 104 IN 102 IN 105 IN 106 OUT OUT OUT OUT OUT OUT	6 101 102 105 103 104 106

输入示例

输出示例

2
5 2501 2502 2503 2504 2505
6 26001 26002 26003 26004 26005 26006
14
IN 2501
IN 26001
IN 2502
IN 2503
IN 2504
IN 2505
OUT
OUT
IN 2602
IN 2603
OUT
OUT
OUT
OUT

6
2501
2502
2503
2504
2505
26001

思路

　　定义一个存team的队列qq，其中，每个元素team是一个队列。也就是说，定义一个存队列的队列。

　　但是由于qq要支持随机访问，故用数组模拟队列。qq中的每个元素team用std::queue或者数组模拟都可以。

　　由于操作数较多(1<=q<=200000) ，若每次操作依据id查找所属team，再查找team再queue中的位置会消耗较多时间，

　　所以用数组teamid存对应id所属的team编号(从1开始)，用数组teamindex存对应team在qq中的index

　　有新人入队时，先检测他所属的team，若在qq中找到，则push在team末尾；若找不到，则在qq中新加一个team，把这个id加入这个team

　　出队时，对qq中的第1个team实行pop操作，再判断该team是否为空，若为空，对qq pop该team

code

 #include <stdio.h>

 #include <stdlib.h>

 #include <string.h>

 #include <iostream>

 using namespace std;

 #include <queue>

 #define MAXID 1000003

 #define MAXTEAM 1003

 #define MAXOP 200003

 int teamid[MAXID] = {};//teamid[id]

 bool inque[MAXID] = {}; //inque[id]

 int teamindex[MAXTEAM] = {};//teamindex[teamid[id]]

 queue<int> ans;

 queue<int> qq[MAXOP];

 int main()

 {

     int i,j,u;

     int t,id,op;

     char str[]={};

     scanf("%d",&t);

     for(i=;i<=t;i++)

     {

         scanf("%d",&u);

         for(j=;j<=u;j++)

         {

             scanf("%d",&id);

             teamid[id] = i;

         }

     }

     int l=,r=;

     scanf("%d",&op);

     getchar();

     for(i=;i<=op;i++)

     {

         scanf("%s",str);

         if(strcmp(str,"IN")==)

         {

             scanf("%d",&id);

             getchar();

             if(inque[id]==false)

             {

                 inque[id] = true;

                 if(teamindex[teamid[id]]==)

                 {

                     ++r;

                     qq[r].push(id);

                     teamindex[teamid[id]] = r;

                 }

                 else

                 {

                     qq[teamindex[teamid[id]]].push(id);

                 }

             }

         }

         else if(strcmp(str,"OUT")==)

         {

             if(l<=r)

             {

                 inque[qq[l].front()] = false;

                 ans.push(qq[l].front());

                 int tmp = qq[l].front();

                 qq[l].pop();

                 if(qq[l].empty())

                 {

                     teamindex[teamid[tmp]] = ;

                     ++l;

                 }

             }

         }

     }

     printf("%d\n",ans.size());

     while(!ans.empty())

     {

         printf("%d\n",ans.front());

         ans.pop();

     }

     return ;

 }

之前贴的代码有bug (新代码已修正)：

　　（1）在OUT操作时应判断 (l<=r)；

　　（2）qq数组定义过小，重新定义为 queue<int> qq[MAXOP];，其中 MAXOP = 200003

　　　　注意，qq数组最好在全局定义，不然OJ上会SO

巴特西

数据结构 queue

最新文章

热门文章