bzoj1036:[Zjoi2008]树的统计

Description

 

 

一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。 
我们将以下面的形式来要求你对这棵树完成一些操作： 
I. CHANGE u t : 把结点u的权值改为t 
II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 
III. QSUM u v: 询问从点u到点v的路径上的节点的权值和 
注意：从点u到点v的路径上的节点包括u和v本身 

Input

输入文件的第一行为一个整数n，表示节点的个数。 
接下来n – 1行，每行2个整数a和b，表示节点a和节点b之间有一条边相连。 
接下来n行，每行一个整数，第i行的整数wi表示节点i的权值。 
接下来1行，为一个整数q，表示操作的总数。 
接下来q行，每行一个操作，以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。 
保证1<=n<=30000，0<=q<=200000；中途操作中保证每个节点的权值w在-30000到30000之间。 

Output

对于每个“QMAX”或者“QSUM”的操作，每行输出一个整数表示要求输出的结果。

Sample Input

4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4

Sample Output

4
1
2
2
10
6
5
6
5
16

现在很晚了，这种板子题我也不知道怎么表述。

 

到时候再补吧，先丢代码

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=60001;
int n,m,cnt,now,son[N],f[N],nex[N],h[N],top[N],id[N],size[N],dep[N],ans,mxx;
struct oo{int a,b,v,sum,mx;}s[N*2-20000];
void ins(int x,int y)
{
    son[++cnt]=y;
    nex[cnt]=h[x];
    h[x]=cnt;
}
void dfs(int x)
{
    size[x]=1;
    for(int i=h[x];i;i=nex[i])
    {
        int v=son[i];
        if(v==f[x])continue;
        dep[v]=dep[x]+1;
        f[v]=x;
        dfs(v);
        size[x]+=size[v];
    }
}
void dfs2(int x,int y)
{
    int k=0;
    id[x]=++now;top[x]=y;
    for(int i=h[x];i;i=nex[i])
        if(dep[son[i]]>dep[x]&&size[son[i]]>size[k])
            k=son[i];
    if(k==0)return ;
    dfs2(k,y);
    for(int i=h[x];i;i=nex[i])
    {
        if(son[i]==k)continue;
        if(dep[son[i]]>dep[x])
            dfs2(son[i],son[i]);
    }
}
void build(int x,int l,int r)
{
    s[x].a=l,s[x].b=r;
    if(l==r){s[x].mx=-9999999;return ;}
    build(x<<1,l,l+r>>1);
    build(x<<1|1,(l+r>>1)+1,r);
}
void change(int x,int l,int v)
{
    if(s[x].a==s[x].b){s[x].v=v;s[x].sum=v;s[x].mx=v;return ;}
    int mid=s[x].a+s[x].b>>1;
    if(l<=mid)change(x<<1,l,v);
    else change(x<<1|1,l,v);
    s[x].mx=max(s[x<<1].mx,s[x<<1|1].mx);
    s[x].sum=s[x<<1].sum+s[x<<1|1].sum;
}
void get(int x,int l,int r)
{
    if(l<=s[x].a&&r>=s[x].b)
        ans+=s[x].sum,mxx=max(mxx,s[x].mx);
    else
    {
        int mid=s[x].a+s[x].b>>1;
        if(l<=mid)get(x<<1,l,r);
        if(r>mid)get(x<<1|1,l,r);
    }
}
void qmax(int x,int y)
{
    int mx=-99999999;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        mxx=-99999999;ans=0;
        get(1,id[top[x]],id[x]);
        mx=max(mx,mxx);
        x=f[top[x]];
    }
    if(id[x]>id[y])swap(x,y);
    mxx=-99999999;ans=0;
    get(1,id[x],id[y]);
    mx=max(mx,mxx);
    printf("%d\n",mx);
}
void qsum(int x,int y)
{
    int sum=0;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        ans=0;
        get(1,id[top[x]],id[x]);
        sum+=ans;
        x=f[top[x]];
    }
    if(id[x]>id[y])swap(x,y);
    ans=0;
    get(1,id[x],id[y]);
    sum+=ans;
    printf("%d\n",sum);
}
int main()
{
    scanf("%d",&n);
    for(int i=1,x,y;i<n;i++)
        scanf("%d%d",&x,&y),ins(x,y),ins(y,x);
    dfs(1),dfs2(1,1);build(1,1,n);
    for(int i=1,x;i<=n;i++)
        scanf("%d",&x),change(1,id[i],x);
    scanf("%d",&m);char p[13];int a,b;
    while(m--)
    {
        scanf("%s",p+1);
        if(p[2]=='S')
        {
            scanf("%d%d",&a,&b);
            qsum(a,b);
        }
        if(p[2]=='M')
        {
            scanf("%d%d",&a,&b);
            qmax(a,b);
        }
        if(p[2]=='H')
        {
            scanf("%d%d",&a,&b);
            change(1,id[a],b);
        }
    }
}