【SYZOJ279】滑稽♂树（树套树）

题面

SYZOJ

CJOJ

题目描述

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。

输出格式

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

样例

input

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

output

4

4

5

2

7

题解

树套树大裸题

直接来一个带修改主席树就没了

真简单

#include<iostream>

#include<cstdio>

#include<cstdlib>

#include<cstring>

#include<cmath>

#include<algorithm>

#include<set>

#include<map>

#include<vector>

#include<queue>

using namespace std;

#define ll long long

#define RG register

#define MAX 55555

inline int read()

{

    RG int x=0,t=1;RG char ch=getchar();

    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();

    if(ch=='-')t=-1,ch=getchar();

    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();

    return x*t;

}

int n,W[MAX];

struct Line{int v,next;}e[MAX<<1];

int h[MAX],cnt=1;

inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}

int dfn[MAX],low[MAX],tim;

void dfs(int u,int ff)

{

	dfn[u]=++tim;

	for(int i=h[u];i;i=e[i].next)

		if(e[i].v!=ff)dfs(e[i].v,u);

	low[u]=tim;

}

int rt[MAX],tot;

struct Node{int ls,rs,v;}t[MAX<<7];

int lb(int x){return x&(-x);}

void Modify(int &x,int ff,int l,int r,int p,int w)

{

	x=++tot;t[x]=t[ff];t[x].v+=w;

	if(l==r)return;

	int mid=(l+r)>>1;

	if(p<=mid)Modify(t[x].ls,t[ff].ls,l,mid,p,w);

	else Modify(t[x].rs,t[ff].rs,mid+1,r,p,w);

}

void preModify(int x,int p,int w)

{

	for(int i=x;i<=n;i+=lb(i))

		Modify(rt[i],rt[i],1,10000,p,w);

}

int tmp1[MAX],tmp2[MAX],t1,t2;

void preQuery(int l,int r)

{

	t1=t2=0;

	for(int i=l;i;i-=lb(i))tmp1[++t1]=rt[i];

	for(int i=r;i;i-=lb(i))tmp2[++t2]=rt[i];

}

int Query(int l,int r,int K)

{

	if(l==r)return l;

	int s=0,mid=(l+r)>>1;

	for(int i=1;i<=t1;++i)s-=t[t[tmp1[i]].ls].v;

	for(int i=1;i<=t2;++i)s+=t[t[tmp2[i]].ls].v;

	if(s>=K)

	{

		for(int i=1;i<=t1;++i)tmp1[i]=t[tmp1[i]].ls;

		for(int i=1;i<=t2;++i)tmp2[i]=t[tmp2[i]].ls;

		return Query(l,mid,K);

	}

	else

	{

		for(int i=1;i<=t1;++i)tmp1[i]=t[tmp1[i]].rs;

		for(int i=1;i<=t2;++i)tmp2[i]=t[tmp2[i]].rs;

		return Query(mid+1,r,K-s);

	}

}

int QuerySum(int l,int r,int L,int R)

{

	int ret=0;

	if(L<=l&&r<=R)

	{

		for(int i=1;i<=t1;++i)ret-=t[tmp1[i]].v;

		for(int i=1;i<=t2;++i)ret+=t[tmp2[i]].v;

		return ret;

	}

	int mid=(l+r)>>1;

	if(L<=mid)

	{

		int T1[MAX],T2[MAX];

		for(int i=1;i<=t1;++i)T1[i]=tmp1[i],tmp1[i]=t[tmp1[i]].ls;

		for(int i=1;i<=t2;++i)T2[i]=tmp2[i],tmp2[i]=t[tmp2[i]].ls;

		ret+=QuerySum(l,mid,L,R);

		for(int i=1;i<=t1;++i)tmp1[i]=T1[i];

		for(int i=1;i<=t2;++i)tmp2[i]=T2[i];

	}

	if(R>mid)

	{

		int T1[MAX],T2[MAX];

		for(int i=1;i<=t1;++i)T1[i]=tmp1[i],tmp1[i]=t[tmp1[i]].rs;

		for(int i=1;i<=t2;++i)T2[i]=tmp2[i],tmp2[i]=t[tmp2[i]].rs;

		ret+=QuerySum(mid+1,r,L,R);

		for(int i=1;i<=t1;++i)tmp1[i]=T1[i];

		for(int i=1;i<=t2;++i)tmp2[i]=T2[i];

	}

	return ret;

}

int main()

{

	n=read();

	for(int i=1;i<=n;++i)W[i]=read();

	for(int i=1;i<n;++i)

	{

		int u=read(),v=read();

		Add(u,v);Add(v,u);

	}

	dfs(1,0);

	for(int i=1;i<=n;++i)

		preModify(dfn[i],W[i],1);

	int Q=read();

	while(Q--)

	{

		int opt=read(),u=read();

		if(opt==1)

		{

			preQuery(dfn[u]-1,low[u]);

			printf("%d\n",Query(1,10000,read()));

		}

		if(opt==2)

		{

			int l=read(),r=read();

			preQuery(dfn[u]-1,low[u]);

			printf("%d\n",QuerySum(1,10000,l,r));

		}

		if(opt==3)

		{

			preModify(dfn[u],W[u],-1);

			W[u]=read();

			preModify(dfn[u],W[u],1);

		}

	}

	return 0;

}