BZOJ1036: [ZJOI2008]树的统计Count

1036: [ZJOI2008]树的统计Count

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 5084  Solved: 2156
[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

树的分治

题解：

原来树链剖分水过。

LCT大法好，不用码SGT了。。。注意access的时候也要pushup

代码：

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<string>
#define inf 1000000000
#define maxn 30000+1000
#define maxm 500+100
#define eps 1e-10
#define ll long long
#define pa pair<int,int>
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}
    return x*f;
}
struct edge{int go,next;}e[2*maxn];
int n,m,tot,head[maxn],fa[maxn],c[maxn][2],sta[maxn],next[maxn],mx[maxn],sum[maxn],v[maxn];
bool rev[maxn];
inline void insert(int x,int y)
{
    e[++tot].go=y;e[tot].next=head[x];head[x]=tot;
    e[++tot].go=x;e[tot].next=head[y];head[y]=tot;
}
inline bool isroot(int x)
{
    return c[fa[x]][0]!=x&&c[fa[x]][1]!=x;
}
inline void pushup(int x)
{
    mx[x]=max(v[x],max(mx[c[x][0]],mx[c[x][1]]));
    sum[x]=sum[c[x][0]]+sum[c[x][1]]+v[x];
}
void rotate(int x)
{
    int y=fa[x],z=fa[y],l=(c[y][1]==x),r=l^1;
    if(!isroot(y))c[z][c[z][1]==y]=x;
    fa[x]=z;fa[y]=x;fa[c[x][r]]=y;
    c[y][l]=c[x][r];c[x][r]=y;
    pushup(y);pushup(x);
}
void pushdown(int x)
{
    if(!rev[x])return;
    rev[x]^=1;rev[c[x][0]]^=1;rev[c[x][1]]^=1;
    swap(c[x][0],c[x][1]);
}
void splay(int x)
{
    int top=0;sta[++top]=x;
    for(int y=x;!isroot(y);y=fa[y])sta[++top]=fa[y];
    for(;top;)pushdown(sta[top--]);
    while(!isroot(x))
    {
        int y=fa[x],z=fa[y];
        if(!isroot(y))
         {
             if(c[z][0]==y^c[y][0]==x)rotate(x);else rotate(y);
         }
        rotate(x); 
    }
    
}
void access(int x)
{
    for(int y=0;x;x=fa[x])
    {
        splay(x);c[x][1]=y;pushup(x);y=x;
    }
}
void makeroot(int x)
{
    access(x);splay(x);rev[x]^=1;
}
int find(int x)
{
    access(x);splay(x);
    while(c[x][0])x=c[x][0];
    return x;
}
void link(int x,int y)
{
    makeroot(x);fa[x]=y;splay(x);
}
void cut(int x,int y)
{
    makeroot(x);access(y);splay(y);c[y][0]=fa[x]=0;
}
int main()
{
    freopen("input.txt","r",stdin);
    freopen("output.txt","w",stdout);
    n=read();
    int x,y,top=0;char ch[10];
    for(int i=1;i<n;i++)x=read(),y=read(),insert(x,y);
    for(int i=1;i<=n;i++)mx[i]=sum[i]=v[i]=read();
    sta[++top]=1;
    for(int i=1;i<=top;i++)
     for(int k=sta[i],j=head[k];j;j=e[j].next)
      if(e[j].go!=fa[k])
       {
           fa[y=e[j].go]=k;sta[++top]=y;
       }  
    mx[0]=-inf;
    m=read();
    while(m--)
    {
        scanf("%s",ch);x=read();y=read();
        if(ch[0]=='C')
        {
            makeroot(x);v[x]=y;pushup(x);
        }
        else
        {
            makeroot(x);
            access(y);splay(y);printf("%d
",ch[1]=='M'?mx[y]:sum[y]);
        }
    }   
    return 0;
}