ZCMU-1396-队列

1396: 队列问题(2)

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 55   Solved: 14
[ Submit][ Status][ Web Board]

Description

有一个含有n个元素的队列q，每个元素的大小满足1<=xi<=9(0<i<n)。队列有一种操作，对于队首元素若是整个队列最大的则出队列，否则加入队尾。对于一个给定的m，你能告诉我xm是第几个出队列的吗？

Input

输入数据第一行是一个整数T(1<=T<=1000)，表示输入数据的组数；每组数据的第一行是两正整数n表示队列的大小和第几个元素(1<n<=1000，0<=m<n)，第二行有n个数xi ，分别代表每个元素的大小。

Output

对于每组测试数据，输出xm是第几个出队列。

Sample Input

3

1 0

5

4 2

1 2 3 4

6 0

1 1 9 1 1 1

Sample Output

1

2

5

【解析】

这里又让我学到知识了multset是允许可以有重复键值的set，而set是不能允许的，还有就是我们的greater<int>是

一个重载函数，可以这么用sort(a,a+n,greater<int>)这样的话就是从大到小排序了，如果是less<int>那就是从小

到大了。还有就是set怎么删除，queue怎么用需要注意的。还有个借鉴之处就是加个-1标记已经弹出了。

#include <cstdio>
#include <iostream>
#include <queue>
#include <set>
using namespace std;
queue<int>a;
multiset<int,greater<int> >b;//可以有重复的set
int main()
{
	int t,count,p,i;
	scanf("%d",&t);
	while (t--)
	{
	    count=0;
		int n,m;
		scanf("%d%d",&n,&m);
		while (!a.empty())
            a.pop();//清空queue不断的弹出元素
		b.clear();
		for (i=0;i<n;i++)
		{
           scanf("%d",&p);
			a.push(p);
			b.insert(p);
			if (i==m)
                a.push(-1);
		}
		while (1)
		{
			if (a.front()==*b.begin())
			{
				count++;
				a.pop();
				b.erase(b.begin());
				if (a.front()==-1) break;//出循环了
			}
			else
			{
				a.push(a.front());//继续加入到队列当中放到队尾
				a.pop();
			}
		}
		printf("%d\n",count);
	}
	return 0；
}