P1533 可怜的狗狗

思路

终于做了一个简单题,233
先对区间拍个序
因为他说所有区间互不包含
所以排序之后可以双指针l,r指呀指
容易得到 这样每个值最多插入删除一边
一次操作logn
总复杂度nlogn

代码

#include <bits/stdc++.h>
#define FOR(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
const int maxn=3e5+7;
int read() {
	int x=0,f=1;char s=getchar();
	for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;
	for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';
	return x*f;
}
int n,m,rt,a[maxn];
int ch[maxn][2],val[maxn],pri[maxn],siz[maxn],cnt;
struct node {
	int l,r,k,id;
	bool operator < (const node &a) const {
		return l<a.l;
	}
}Q[maxn];
int ans[maxn];
void pushup(int x) {
	siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+1;
}
int new_edge(int a) {
	val[++cnt]=a;
	pri[cnt]=rand();
	siz[cnt]=1;
	return cnt;
}
int merge(int x,int y) {
	if(!x||!y) return x+y;
	if(pri[x]<pri[y]) {
		ch[x][1]=merge(ch[x][1],y);
		pushup(x);return x;
	} else {
		ch[y][0]=merge(x,ch[y][0]);
		pushup(y);return y;
	}
}
void split(int now,int k,int &x,int &y) {
	if(!now) x=y=0;
	else {
		if(val[now]<=k)
			x=now,split(ch[now][1],k,ch[x][1],y);
		else
			y=now,split(ch[now][0],k,x,ch[y][0]);
		pushup(now);
	}
}
void insert(int a) {
	int x,y;
	split(rt,a,x,y);
	rt=merge(merge(x,new_edge(a)),y);
}
void delet(int a) {
	int x,y,z;
	split(rt,a,x,z);
	split(x,a-1,x,y);
	y=merge(ch[y][0],ch[y][1]);
	rt=merge(merge(x,y),z);
}
int k_th(int now,int k) {
	while(233) {
		if(siz[ch[now][0]]+1==k) return val[now];
		if(siz[ch[now][0]]>=k) now=ch[now][0];
		else k-=siz[ch[now][0]]+1,now=ch[now][1];
	}
}
int main() {
	n=read(),m=read();
	FOR(i,1,n) a[i]=read();
	FOR(i,1,m) Q[i].l=read(),Q[i].r=read(),Q[i].k=read(),Q[i].id=i;
	sort(Q+1,Q+1+m);
	int l=1,r=0;
	FOR(i,1,m) {
		while(r!=Q[i].r) insert(a[++r]); 
		while(l!=Q[i].l) delet(a[l++]);
		ans[Q[i].id]=k_th(rt,Q[i].k); 
	}
	FOR(i,1,m) cout<<ans[i]<<"
";
	return 0;
}

原文地址:https://www.cnblogs.com/dsrdsr/p/10092081.html