[SYZOI Round1]滑稽♂树

【题目描述】

zzsyz实验楼里面种了一棵滑稽树，只有滑稽之力达到大乘期的oier才能看到。虽然我们看不到，但是还是知道一些信息：

这真的是一棵树，由n个节点，n-1条边联通。一号滑稽果同时也是整棵滑稽树的树根。滑稽树上每个节点有一个滑稽果，每个滑稽果有它的重量。

雪甜甜公主是神犇当然看得到那棵滑稽树啦，现在她感兴趣的是这样三件事

1：滑稽树太大啦，雪甜甜公主有的时候只想知道，在以某一个滑稽果为根的子滑稽树里面，重量第k小的果子的重量是多少？

2：除了重量第k小的果子，雪甜甜还想知道以某个滑稽果为根的子滑稽树里面，重量在[a, b]这个范围内的滑稽果有多少个。

3：雪甜甜还喜欢吃滑稽果，但是吃完，原来滑稽果的位置上还会长出一个新的滑稽果，只是重量可能不一样。

【输入格式】

第一行一个正整数n，表示滑稽树有n个节点。

第二行n个正整数，分别描述1号，2号,,,，n号节点滑稽果的重量。

接下来n-1行，每行2个正整数u, v ∈ [1, n]，表示滑稽果u与滑稽果v之间有树枝连接。

接下来一个正整数q，表示雪甜甜有q次行动

之后q行，有这样3种形式

1 u k 雪甜甜公主询问以u为根的子滑稽树中，重量第k小的滑稽果的重量。

2 u a b 雪甜甜公主想知道，以u为根的子滑稽树中，重量在[a, b]范围内的滑稽果有多少个。

3 u x 雪甜甜公主吃掉了编号为u的滑稽果，但是在原位置上立刻长出来了一个重量为x的滑稽果。因为位置没有变，所以编号还是u。

【输出格式】

对于每次询问，输出结果。

【样例输入】

5



3 4 6 1 2



1 2



1 3



3 4



3 5



7



1 1 4



2 1 1 5



3 4 5



1 1 4



2 3 3 6



3 5 7



1 3 3

【样例输出】

4



4



5



2



7
【数据范围以及提示】

N：
对于前35%的数据满足，N <= 5000
对于前50%的数据满足，N <= 10000
对于前100%的数据满足，N <= 30000
滑稽果的重量：对于100%d的数据满足 滑稽果的重量 <= 10000
询问：询问的个数Q：
对于前50%的数据满足 Q <= 10000
对于前100%的数据满足 Q <= 50000
对于前25%的数据，只有第一种询问。
对于前65%的数据，有第1，2种询问。
对于100%的数据第1，2，3种询问都存在。
对于前35%的数据，满足一个特殊的限制条件：每次询问的滑稽果u = 1保证询问k小重量的滑稽果的时候，k值∈ [1, 子树的节点数]
来自 http://syzoj.com/problem/279
树状数组套dfs序主席树，比较裸
代码：
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
const int maxn=200000+10;
vector<int>A[maxn];
int w[maxn];
int root[maxn*50],ls[maxn*50],rs[maxn*50];
int sum[maxn*50];
int L[110],R[110];
int tid[maxn],size[maxn];
int n,m;
int sz=0,ti=0;
int c,d;
inline int lowbit(int x){
	return x&(-x);
}
inline void insert(int &x,int y,int l,int r,int c,int v){
	x=++sz;
	sum[x]=sum[y]+v;
	if(l==r)
		return ;
	ls[x]=ls[y];
	rs[x]=rs[y];
	int mid=(l+r)>>1;
	if(mid>=c)
		insert(ls[x],ls[y],l,mid,c,v);
	else insert(rs[x],rs[y],mid+1,r,c,v);
}
inline int query1(int l,int r,int ql,int qr){
	int a[110],b[110];
	int suml=0,sumr=0;
	for(int i=1;i<=c;i++)
		suml+=sum[L[i]];
	for(int i=1;i<=d;i++)
		sumr+=sum[R[i]];
	for(int i=1;i<=c;i++)
		a[i]=L[i];
	for(int i=1;i<=d;i++)
		b[i]=R[i];
	if(l>=ql&&r<=qr)
		return sumr-suml;
	if(l==r)
		return 0;
	int mid=(l+r)>>1;
	int ans=0;
	if(mid>=ql){
		for(int i=1;i<=c;i++)
			L[i]=ls[L[i]];
		for(int i=1;i<=d;i++)
			R[i]=ls[R[i]];
		ans+=query1(l,mid,ql,qr);
		for(int i=1;i<=c;i++)
			L[i]=a[i];
		for(int i=1;i<=d;i++)
			R[i]=b[i];
	}
	if(mid<qr){
		for(int i=1;i<=c;i++)
			L[i]=rs[L[i]];
		for(int i=1;i<=d;i++)
			R[i]=rs[R[i]];
		ans+=query1(mid+1,r,ql,qr);
		for(int i=1;i<=c;i++)
			L[i]=a[i];
		for(int i=1;i<=d;i++)
			R[i]=b[i];
	}
	return ans;
}
inline int query2(int l,int r,int k){
	int a[110],b[110];
	int suml=0,sumr=0;
	for(int i=1;i<=c;i++)
		suml+=sum[ls[L[i]]];
	for(int i=1;i<=d;i++)
		sumr+=sum[ls[R[i]]];
	for(int i=1;i<=c;i++)
		a[i]=L[i];
	for(int i=1;i<=d;i++)
		b[i]=R[i];
	if(l==r)
		return l;
	int mid=(l+r)>>1;
	if(sumr-suml>=k){
		for(int i=1;i<=c;i++)
			L[i]=ls[L[i]];
		for(int i=1;i<=d;i++)
			R[i]=ls[R[i]];
		return query2(l,mid,k);
		for(int i=1;i<=c;i++)
			L[i]=a[i];
		for(int i=1;i<=d;i++)
			R[i]=b[i];
	}
	else {
		for(int i=1;i<=c;i++)
			L[i]=rs[L[i]];
		for(int i=1;i<=d;i++)
			R[i]=rs[R[i]];
		return query2(mid+1,r,k-sumr+suml);
		for(int i=1;i<=c;i++)
			L[i]=a[i];
		for(int i=1;i<=d;i++)
			R[i]=b[i];
	}
}
inline void dfs(int x,int f){
	tid[x]=++ti;
	size[x]=1;
	for(int i=0;i<A[x].size();i++){
		int u=A[x][i];
		if(u==f)
			continue;
		dfs(u,x);
		size[x]+=size[u];
	}
}
int main(){
	freopen("hjtree.in","r",stdin);
	freopen("hjtree.out","w",stdout);
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&w[i]);
	int x,y;
	for(int i=1;i<n;i++){
		scanf("%d %d",&x,&y);
		A[x].push_back(y);
		A[y].push_back(x);
	}
	dfs(1,0);
	for(int i=1;i<=n;i++)
		for(int j=tid[i];j<=n;j+=lowbit(j))
			insert(root[j],root[j],1,10000,w[i],1);
	scanf("%d",&m);
	int z,p;
	//for(int i=1;i<=sz;i++)
	//	printf("%d\n",sum[i]);
	while(m--){
		scanf("%d",&x);
		if(x==1){
			scanf("%d %d",&y,&z);
			c=d=0;
			for(int i=tid[y]-1;i>=1;i-=lowbit(i))
				L[++c]=root[i];
			for(int i=tid[y]+size[y]-1;i>=1;i-=lowbit(i))
				R[++d]=root[i];
			printf("%d\n",query2(1,10000,z));
		}
		else if(x==2){
			scanf("%d %d %d",&y,&z,&p);
			c=d=0;
			for(int i=tid[y]-1;i>=1;i-=lowbit(i))
				L[++c]=root[i];
			for(int i=tid[y]+size[y]-1;i>=1;i-=lowbit(i))
				R[++d]=root[i];
			printf("%d\n",query1(1,10000,z,p));
		}
		else {
			scanf("%d %d",&y,&z);
			for(int i=tid[y];i<=n;i+=lowbit(i))
				insert(root[i],root[i],1,10000,w[y],-1);
			for(int i=tid[y];i<=n;i+=lowbit(i))
				insert(root[i],root[i],1,10000,z,1);
			w[y]=z;
		}
	}
return 0;
}