【洛谷】2464 [SDOI2008]郁闷的小J 平衡树

22 篇文章 0 订阅
10 篇文章 0 订阅

题目传送门

这题的思路好诡异啊……

最近在看有关平衡树的一些东西,看到书上的课后习题有这题。

看到这题我就傻了——这TMD怎么用平衡树做?

题解给出的思路是对每本书的编号离散化后建立平衡树森林,权值为这本书所在的位置。

对于每个修改就把这本书从原来的平衡树中删掉,然后插入新的平衡树中。

询问非常的简单,差分一下就行了。

(话说这题如果用分块做感觉要好想多了……)

附上AC代码:

#include <cstdio>
#include <cctype>
#include <algorithm>
using namespace std;

const int N=2e5+10;
struct query{
	int x,y,w;
}q[N];
int n,m,a[N],s[N],num,len,bh[N],t,sy[N],rt[N],f[N],w[N],sz[N],size,ch[N][2];
char c;

inline char nc(void){
	static char ch[100010],*p1=ch,*p2=ch;
	return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
	static char c=nc();int f=1;
	for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
	for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());
	a*=f;return;
}

inline int find(int x){
	int l=1,r=len,mid,ret=0;
	while (l<=r)
		if (bh[(mid=(l+r)>>1)]<=x) ret=mid,l=mid+1;
		else r=mid-1;
	return ret;
}

inline void updata(int x){sz[x]=sz[ch[x][0]]+sz[ch[x][1]]+1;}

inline void rotate(int x){
	int y=f[x],op=(ch[y][1]==x);
	ch[y][op]=ch[x][op^1];
	if (ch[x][op^1]) f[ch[x][op^1]]=y;
	f[x]=f[y];
	if (f[y]) ch[f[y]][ch[f[y]][1]==y]=x;
	f[y]=x,ch[x][op^1]=y;
	return updata(y),updata(x);
}

inline void splay(int x,int ed,int wz){
	for (int fa=f[x]; fa!=ed; rotate(x),fa=f[x])
		if (f[fa]!=ed) rotate((x==ch[fa][0])==(fa==ch[f[fa]][0])?fa:x);
	if (!ed) rt[wz]=x;
	return;
}

inline void ist(int wz,int x){
	int now=rt[wz];
	while (ch[now][w[now]<x]) now=ch[now][w[now]<x];
	w[++size]=x,f[size]=now,ch[size][0]=ch[size][1]=0;
	if (now) ch[now][w[now]<x]=size;
	return splay(size,0,wz);
}

inline void del(int wz,int x){
	int now=rt[wz];
	while (w[now]!=x) now=ch[now][w[now]<x];
	splay(now,0,wz);
	if (!ch[rt[wz]][0]) {f[rt[wz]=ch[rt[wz]][1]]=0; return;}
	now=ch[rt[wz]][0];
	while (ch[now][1]) now=ch[now][1];
	splay(now,rt[wz],wz);
	f[ch[now][1]=ch[rt[wz]][1]]=now,f[rt[wz]=now]=0;
	return;
}

inline int ask(int wz,int x){
	int now=rt[wz],ans=0;
	while (now)
		if (w[now]>x) ans=now,now=ch[now][0];
		else now=ch[now][1];
	splay(ans,0,wz);
	return sz[ch[rt[wz]][0]];
}

int main(void){
	read(n),read(m);
	for (int i=1; i<=n; ++i) read(a[i]),s[i]=a[i];
	num=n;
	for (int i=1; i<=m; ++i){
		c=nc();
		while (c!='Q'&&c!='C') c=nc();
		read(q[i].x),read(q[i].y);
		switch (c){
			case 'C': s[++num]=q[i].y; break;
			case 'Q': read(q[i].w); break;
		}
	}
	sort(s+1,s+1+num);
	for (int i=1; i<=num; ++i) if (s[i]!=s[i+1]) bh[++len]=s[i];
	for (int i=1; i<=len; ++i) ist(i,2e9);
	for (int i=1; i<=n; ++i){
		sy[i]=(t=find(a[i]));
		ist(t,i);
	}
	for (int i=1; i<=m; ++i)
		if (q[i].w){
			t=find(q[i].w);
			printf("%d\n",ask(t,q[i].y)-ask(t,q[i].x-1));
		}
		else{
			t=find(q[i].y);
			del(sy[q[i].x],q[i].x);
			sy[q[i].x]=t;
			ist(t,q[i].x);
		}
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值