2016 UESTC Training for Data Structures I - 郭大侠与线上游戏 CDOJ 1339 pb_ds黑科技

I - 郭大侠与线上游戏

就是给你一个队列,支持队尾插入和队头弹出,或者询问当前队列里的数的中位数

别人跟我讲stl黑科技能过,然后我就套pb_ds,就过了

这个pb_ds只在linux下可以,但是评测姬一般都是linux下的,所以都能用,本地用就用mingw,然后我用的是Tree-Like Container,可以查询区间第k大,然后k就取中位数就好了

 

如果不用这个黑科技的话,可以用两个set/优先队列/堆来维护,具体的做法可以百度 堆求中位数


代码:

#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
using namespace __gnu_pbds;
#define maxn 1000005
typedef tree, rb_tree_tag> Tree;
typedef tree, rb_tree_tag,
tree_order_statistics_node_update> set_t;
int a[maxn];
int size,pos;
int main() {
	set_t s;
	int n;
	scanf("%d", &n);
	int op, x, k;
	for (int i = 0; i < n; ++i)
	{
		scanf("%d", &op);
		if (op == 1)
		{
			scanf("%d", &x);
			a[size++] = x;
			s.insert(x);
		}
		else if (op == 2)
		{
			s.erase(a[pos]);
			++pos;
		}
		else if (op == 3)
		{
			k = (size - pos) / 2;
			printf("%d\n", *s.find_by_order(k));
		}
	}
	return 0;
}

你可能感兴趣的:(数据结构,CDOJ)