bzoj1036 [ZJOI2008]树的统计Count

传送门
Description

　　一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成
一些操作： I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I
II. 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”的形式给出。
对于100％的数据，保证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

题解

树剖练习题。
结果一开始写残了，在修改的时候忘记用pos[x]，结果得了20分。。【数据是不是有一点弱？

CODE：

#include<cstdio>
const int INF=1e9;
const int N=5e4;
struct edge
{
    int next,to;
}a[N<<1];
struct tree
{
    int num,maxn;
}t[N<<2];
char S[10];
int head[N],size[N],deep[N],f[N],son[N],top[N],pos[N],num[N],s[N];
int n,m,x,y,e_num,tot;

inline int max(int a,int b){return a>b?a:b;}
inline int min(int a,int b){return a<b?a:b;}
inline void swap(int &a,int &b){a^=b,b^=a,a^=b;}
inline void read(int &n)
{
    n=0;char c=getchar();bool b=0;
    while(c<'0'||c>'9'){if(c=='-')b=1;c=getchar();}
    while(c>='0'&&c<='9') n=n*10+c-48,c=getchar();
    if(b) n=-n;
}
inline void getstring()
{
    int p=-1;char c=getchar();
    while(c<'A'||c>'Z') c=getchar();
    while(c>='A'&&c<='Z') S[++p]=c,c=getchar();
}
inline void add(int x,int y)
{
    a[++e_num].next=head[x],a[e_num].to=y,head[x]=e_num;
    a[++e_num].next=head[y],a[e_num].to=x,head[y]=e_num;
}
inline void dfs(int now,int fa,int depth)
{
    deep[now]=depth;
    f[now]=fa;
    size[now]=1;
    int tmp=-INF;
    for(int i=head[now];i;i=a[i].next)
      if(a[i].to!=fa)
      {
        dfs(a[i].to,now,depth+1);
        size[now]+=size[a[i].to];
        if(size[a[i].to]>tmp) tmp=size[a[i].to],son[now]=a[i].to;
      }
}
inline void dfs2(int now,int high)
{
    top[now]=high;
    pos[now]=++tot;
    num[tot]=s[now];
    if(son[now]) dfs2(son[now],high);
    for(int i=head[now];i;i=a[i].next)
      if(a[i].to!=f[now]&&a[i].to!=son[now]) dfs2(a[i].to,a[i].to);
}
inline void update(int now)
{
    t[now].num=t[now<<1].num+t[now<<1|1].num;
    t[now].maxn=max(t[now<<1].maxn,t[now<<1|1].maxn);
}
void build(int l,int r,int now)
{
    if(l==r)
    {
        t[now].num=t[now].maxn=num[l];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,now<<1);
    build(mid+1,r,now<<1|1);
    update(now);
}
void Add(int p,int l,int r,int now,int num)
{
    if(l==r)
    {
        t[now].num=t[now].maxn=num;
        return;
    }
    int mid=(l+r)>>1;
    if(p<=mid) Add(p,l,mid,now<<1,num);
    else Add(p,mid+1,r,now<<1|1,num);
    update(now);
}
int ask(int L,int R,int l,int r,int now)
{
    if(L<=l&&r<=R) return t[now].num;
    int mid=(l+r)>>1,ans=0;
    if(L<=mid) ans+=ask(L,R,l,mid,now<<1);
    if(R>mid) ans+=ask(L,R,mid+1,r,now<<1|1);
    return ans;
}
int askmax(int L,int R,int l,int r,int now)
{
    if(L<=l&&r<=R) return t[now].maxn;
    int mid=(l+r)>>1,ans=-INF;
    if(L<=mid) ans=max(ans,askmax(L,R,l,mid,now<<1));
    if(R>mid) ans=max(ans,askmax(L,R,mid+1,r,now<<1|1));
    return ans;
}
int askpath(int x,int y)
{
    int L,R;
    if(top[x]==top[y])
    {
        L=min(pos[x],pos[y]);
        R=max(pos[x],pos[y]);
        return askmax(L,R,1,n,1);
    }
    int ans;
    if(deep[top[x]]<deep[top[y]]) swap(x,y);
    L=min(pos[x],pos[top[x]]);
    R=max(pos[x],pos[top[x]]);
    ans=askmax(L,R,1,n,1);
    return max(ans,askpath(y,f[top[x]]));
}
int asksum(int x,int y)
{
    int L,R;
    if(top[x]==top[y])
    {
        L=min(pos[x],pos[y]);
        R=max(pos[x],pos[y]);
        return ask(L,R,1,n,1);
    }
    int ans;
    if(deep[top[x]]<deep[top[y]]) swap(x,y);
    L=min(pos[x],pos[top[x]]);
    R=max(pos[x],pos[top[x]]);
    ans=ask(L,R,1,n,1);
    return ans+asksum(y,f[top[x]]);
}
int main()
{
    read(n);
    for(int i=1;i<n;i++)
      read(x),read(y),add(x,y);
    for(int i=1;i<=n;i++)
      read(s[i]);
    dfs(1,0,1);
    dfs2(1,1);
    build(1,n,1);
    read(m);
    while(m--)
    {
        getstring();
        read(x),read(y);
        if(S[1]=='H') Add(pos[x],1,n,1,y);
        else if(S[1]=='M') printf("%d\n",askpath(x,y));
        else printf("%d\n",asksum(x,y));
    }
    return 0;
}