【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;
}