树链剖分（基于点权）bzoj1036

1036: [ZJOI2008]树的统计Count

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 6204   Solved: 2569
[ Submit][ Status]

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”的形式给出。 对于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

HINT

Source

树的分治

思路：跟基于边权的基本一样，只不过是把边在线段树中的位置换成点在线段树中的位置

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
const int maxn=30010;
const int INF=1000000000;
struct Edge
{
    int v,next;
}edge[maxn*2];
int top[maxn];
int head[maxn],tot;
int fa[maxn];
int deep[maxn];
int w[maxn];
int fw[maxn];
int son[maxn];
int num[maxn];
int pos;
int N,Q;
int a[maxn];
void init()
{
    tot=0;
    memset(head,-1,sizeof(head));
    pos=0;
    memset(son,-1,sizeof(son));
}
void add_edge(int u,int v)
{
    edge[tot].v=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}
void dfs1(int u,int pre,int d)
{
    fa[u]=pre;
    deep[u]=d;
    num[u]=1;
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        if(v==pre)continue;
        dfs1(v,u,d+1);
        num[u]+=num[v];
        if(son[u]==-1||num[son[u]]<num[v])
            son[u]=v;
    }
}
void dfs2(int u,int sp)
{
    top[u]=sp;
    w[u]=pos++;
    fw[w[u]]=u;
    if(son[u]==-1)return ;
    //if(son[u]!=-1)
        dfs2(son[u],sp);
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        if(v!=son[u]&&v!=fa[u])
            dfs2(v,v);
    }
}
struct IntervalTree
{
    int maxv[maxn<<2];
    int sum[maxn<<2];
    void build(int o,int l,int r)
    {
        sum[o]=maxv[o]=0;
        if(l==r)
        {
            sum[o]=maxv[o]=a[fw[l]];
            return ;
        }
        int mid=(l+r)>>1;
        build(o<<1,l,mid);
        build(o<<1|1,mid+1,r);
        pushup(o);
    }
    void pushup(int o)
    {
        sum[o]=sum[o<<1]+sum[o<<1|1];
        maxv[o]=max(maxv[o<<1],maxv[o<<1|1]);
    }
    void update(int o,int l,int r,int x,int val)
    {
        if(l==r)
        {
            sum[o]=maxv[o]=val;
            return ;
        }
        int mid=(l+r)>>1;
        if(x<=mid)update(o<<1,l,mid,x,val);
        else update(o<<1|1,mid+1,r,x,val);
        pushup(o);
    }
    int queryMax(int o,int l,int r,int q1,int q2)
    {
        if(q1<=l&&r<=q2)return maxv[o];
        int mid=(l+r)>>1;
        int ans=-INF;
        if(q1<=mid)ans=max(ans,queryMax(o<<1,l,mid,q1,q2));
        if(q2>mid)ans=max(ans,queryMax(o<<1|1,mid+1,r,q1,q2));
        return ans;
    }
    int querySum(int o,int l,int r,int q1,int q2)
    {
        if(q1<=l&&r<=q2)return sum[o];
        int mid=(l+r)>>1;
        int ans=0;
        if(q1<=mid)ans+=querySum(o<<1,l,mid,q1,q2);
        if(q2>mid)ans+=querySum(o<<1|1,mid+1,r,q1,q2);
        return ans;
    }
}tree;
int QMAX(int u,int v)
{
    int ans=-INF;
    int f1=top[u],f2=top[v];
    while(f1!=f2)
    {
        if(deep[f1]<deep[f2])
        {
            swap(f1,f2);
            swap(u,v);
        }
        ans=max(ans,tree.queryMax(1,0,pos-1,w[f1],w[u]));
        u=fa[f1];f1=top[u];
    }
    if(deep[u]>deep[v])swap(u,v);
    return max(ans,tree.queryMax(1,0,pos-1,w[u],w[v]));
}
int QSUM(int u,int v)
{
    int ans=0;
    int f1=top[u],f2=top[v];
    while(f1!=f2)
    {
        if(deep[f1]<deep[f2])
        {
            swap(f1,f2);
            swap(u,v);
        }
        ans+=tree.querySum(1,0,pos-1,w[f1],w[u]);
        u=fa[f1];f1=top[u];
    }
    if(deep[u]>deep[v])swap(u,v);
    return ans+tree.querySum(1,0,pos-1,w[u],w[v]);
}
int main()
{
    while(scanf("%d",&N)!=EOF)
    {
        init();
        int u,v;
        for(int i=0;i<N-1;i++)
        {
            scanf("%d%d",&u,&v);
            add_edge(u,v);
            add_edge(v,u);
        }
        for(int i=1;i<=N;i++)scanf("%d",&a[i]);
        dfs1(1,0,0);
        dfs2(1,1);
        tree.build(1,0,pos-1);
        scanf("%d",&Q);
        char op[10];
        while(Q--)
        {
            scanf("%s%d%d",op,&u,&v);
            if(op[0]=='C')
                tree.update(1,0,pos-1,w[u],v);
            else if(!strcmp(op,"QMAX"))
                printf("%d\n",QMAX(u,v));
            else printf("%d\n",QSUM(u,v));
        }
    }
    return 0;
}