bzoj 1036: [ZJOI2008]树的统计Count(树链剖分)

1036: [ZJOI2008]树的统计Count

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 18968   Solved: 7741
[ Submit][ Status][ Discuss]

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

HINT

模板题 没啥好说的 这是树链剖分的详解

http://blog.csdn.net/jaihk662/article/details/73739341

http://www.cnblogs.com/937337156Zhang/p/6047165.html

其实用lca也可以求解

#include <iostream>
#include <cstdio>
#include <cstring>
#include <bits/stdc++.h>
using namespace std;
typedef long long LL ;
const int N = 1e6+10;
const int inf = 2147483647;
vector<int>p[N];
int val[N], cnt, n;
int fa[N],son[N],dep[N],siz[N];
int top[N],rak[N],id[N];
void dfs1(int u,int f)
{
    fa[u]=f,dep[u]=dep[f]+1;
    siz[u]=1;
    for(int i=0; i<p[u].size(); i++)
    {
        int v=p[u][i];
        if(v==f) continue;
        dfs1(v,u);
        siz[u]+=siz[v];
        if(son[u]==0||siz[son[u]]<siz[v])son[u]=v;
    }
    return ;
}
void dfs2(int u,int f)
{
    top[u]=f;
    rak[u]=++cnt;
    id[cnt]=u;
    if(son[u]==0) return ;
    dfs2(son[u],f);
    for(int i=0; i<p[u].size(); i++)
    {
        int v=p[u][i];
        if(v==son[u]||v==fa[u])continue;
        dfs2(v,v);
    }
    return ;
}
void init()
{
    memset(dep,0,sizeof(dep));
    memset(siz,0,sizeof(siz));
    memset(son,0,sizeof(son));
    return ;
}
int sum[N*4],maxt[N*4];
void build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=val[id[l]],maxt[rt]=val[id[l]];
        return ;
    }
    int mid=(l+r)/2;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    maxt[rt]=max(maxt[rt<<1],maxt[rt<<1|1]);
    return ;
}
void update(int x,int v,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=v,maxt[rt]=v;
        return ;
    }
    int mid=(l+r)/2;
    if(x<=mid) update(x,v,l,mid,rt<<1);
    else update(x,v,mid+1,r,rt<<1|1);
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    maxt[rt]=max(maxt[rt<<1],maxt[rt<<1|1]);
    return ;
}
int getmax(int L,int R,int l,int r,int rt)
{
    if(l>=L&&r<=R) return maxt[rt];
    int now=-inf;
    int mid=(l+r)/2;
    if(L<=mid) now=max(now,getmax(L,R,l,mid,rt<<1));
    if(R>mid) now=max(now,getmax(L,R,mid+1,r,rt<<1|1));
    return now;
}
int querymax(int x,int y)
{
    int now=-inf,p1=top[x],p2=top[y];
    while(p1!=p2)
    {
        if(dep[p1]<dep[p2]) swap(x,y),swap(p1,p2);
        now=max(now,getmax(rak[p1],rak[x],1,n,1));
        x=fa[p1];
        p1=top[x];
    }
    if(dep[x]>dep[y]) swap(x,y);
    now=max(now,getmax(rak[x],rak[y],1,n,1));
    return now;
}

int getsum(int L,int R,int l,int r,int rt)
{
    if(l>=L&&r<=R) return sum[rt];
    int now=0;
    int mid=(l+r)/2;
    if(L<=mid) now+=getsum(L,R,l,mid,rt<<1);
    if(R>mid) now+=getsum(L,R,mid+1,r,rt<<1|1);
    return now;
}
int querysum(int x,int y)
{
    int now=0,p1=top[x],p2=top[y];
    while(p1!=p2)
    {
        if(dep[p1]<dep[p2]) swap(x,y),swap(p1,p2);
        now+=getsum(rak[p1],rak[x],1,n,1);
        x=fa[p1];
        p1=top[x];
    }
    if(dep[x]>dep[y]) swap(x,y);
    now+=getsum(rak[x],rak[y],1,n,1);
    return now;
}
char str[20];
int main()
{
    scanf("%d", &n);
    for(int i=0; i<=n; i++) p[i].clear();
    for(int i=1; i<n; i++)
    {
        int a, b;
        scanf("%d %d", &a, &b);
        p[a].push_back(b),p[b].push_back(a);
    }
    for(int i=1; i<=n; i++) scanf("%d", &val[i]);
    init();
    cnt=0;
    dfs1(1,0);
    dfs2(1,1);
    build(1,n,1);
    int q, u, v;
    scanf("%d", &q);
    while(q--)
    {
        scanf("%s",str);
        if(str[0]=='C')
        {
            scanf("%d %d", &u, &v);
            update(rak[u],v,1,n,1);
        }
        else if(str[1]=='M')
        {
            scanf("%d %d", &u, &v);
            printf("%d\n",querymax(u,v));
        }
        else
        {
            scanf("%d %d", &u, &v);
            printf("%d\n",querysum(u,v));
        }
    }
    return 0;
}