1981 如何愉快地与STL玩耍 TFN的 线段树+bitset

题目链接:点击查看

驴蛋蛋在愉快地与STL玩耍

突然间小A跳了出来对驴蛋蛋说,看你与STL玩的很开心啊,那我给你一个大小为N的vector,这个vector上每个位置上是一个set<int>,

每次我会在闭区间 [L,R]

 中的每个set<int>里插入一个数c,或者询问 [L,R] 区间所有set里所有数拿下来排序之后的严格第 K

 小,现在你还开心吗,红红火火恍恍惚惚韩寒会画红槐花!!!!

小A走了,留下驴蛋蛋一个人外加一个长度为 N

 的vector<set<int> >在风中凌乱,你能帮驴蛋蛋解除凌乱吗?

根据小A的 C++2.33

 标准vector可以被视作一个数组,下标从 1 开始,大小不得超过 65536

set<int>是一个集合,其中的数不得超过 10000

 并且会被自动去重

小A最多会进行 Q

 次(不超过 65536 )完全符合 C++2.33 标准的操作,但如果出现询问时 [L,R] 区间里不足 K 个数的情况,你只需要对小A回答 −1

 就好了

输入

第一行两个数字N,Q
接下来Q行每行行行首一个整数op
若op为1,则接下来有3个数L,R,C表示在区间[L,R]中的每个set<int>中插入一个数C(0 <= C <= 10000)
若op为2,则接下来有3个数L,R,K表示查询区间[L,R]所有set<int>中所有数的严格第K小值(去重之后)

输出

输出分一至多行,每行一个整数表示对对应询问的回答

输入样例

5 5
2 4 5 7
1 1 2 8
2 4 5 8
2 4 5 1
2 1 3 1

输出样例

-1
-1
-1
8

题解:

别开4倍!!!!!!!!!!!!!!!!!!!!!!

别pushup!!!!!!!!!!!!!!!!!!!!!

别pushdown!!!!!!!!!!!!!!!!!!!!

少写一点是一点,就过了

#include<bits/stdc++.h>
using namespace std;
const int N=65536;
struct node{
	bitset<10010> s,laz;
}tree[133005];
int n,q;
void update(int pl,int pr,int l,int r,int cur,int val)
{
	tree[cur].s[val]=1;
	if(pl<=l&&r<=pr)
	{
		tree[cur].laz[val]=1;
		return;
	}
	int mid=(r+l)>>1;
	if(pl<=mid) update(pl,pr,l,mid,cur<<1,val);
	if(pr>mid) update(pl,pr,mid+1,r,cur<<1|1,val);
}
bitset<10010> query(int pl,int pr,int l,int r,int cur)
{
	if(pl<=l&&r<=pr)
	{
		return tree[cur].s;
	}
	bitset<10010> res(0);
	res|=tree[cur].laz;
	int mid=(r+l)>>1;
	if(pl<=mid) res|=query(pl,pr,l,mid,cur<<1);
	if(pr>mid) res|=query(pl,pr,mid+1,r,cur<<1|1);
	return res;
}
bitset<10010> p[10010];
int main()
{
	p[0][0]=1;
	for(int i=1;i<=10000;i++)
	{
		p[i]=p[i-1];
		p[i][i]=1;
	}
	scanf("%d%d",&n,&q);
	int op,l,r,c;
	while(q--)
	{
		scanf("%d%d%d%d",&op,&l,&r,&c);
		if(op==1)
		{
			update(l,r,1,n,1,c);
		}
		else
		{
			bitset<10010> ans;
			ans=query(l,r,1,n,1);
			if(ans.count() < c || c==0) printf("-1\n");
			else
			{
				l=0,r=10000;
				op=-1;
				while(l<=r)
				{
					int mid=(r+l)>>1;
					if((p[mid]&ans).count()>=c)
					{
						op=mid;
						r=mid-1;
					}
					else l=mid+1;
				}
				printf("%d\n",op);
			}
		}
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/mmk27_word/article/details/89193574