bzoj1036 树的统计Count

题目：一棵树上有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”的操作，每行输出一个整数表示要求输出的结果。

思路：直接套lct就行了

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000")#include<iostream>#include<algorithm>#include<cstdio>#include<cmath>#include<cstring>#include<string>#include<vector>#include<map>#include<set>#include<queue>#include<stack>#include<list>#include<numeric>using namespace std;#define LL long long#define ULL unsigned long long#define INF 0x3f3f3f3f#define mm(a,b) memset(a,b,sizeof(a))#define PP puts("*********************");template<class T> T f_abs(T a){ return a > 0 ? a : -a; }template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}// 0x3f3f3f3f3f3f3f3f// 0x3f3f3f3fconst int maxn=3e4+50;const int maxe=2*maxn;struct Edge{    int to,next;}edge[maxe];int head[maxn],tot;int ch[maxn][2],key[maxn],pre[maxn],rev[maxn];int Max[maxn],sum[maxn];bool rt[maxn];//标记节点是不是splay的根int n;//n个节点，从1开始计数void init(){    tot=0;    for(int i=0;i<=n;i++){        head[i]=-1;        rt[i]=true;        ch[i][0]=ch[i][1]=0;        pre[i]=0;        rev[i]=0;        sum[i]=0;    }    Max[0]=-2000000000;//0节点一定要初始化好，因为要push_up}void addedge(int u,int v){    edge[tot].to=v;    edge[tot].next=head[u];    head[u]=tot++;}void dfs(int u,int fa){    pre[u]=fa;    Max[u]=key[u];    sum[u]=key[u];    for(int i=head[u];i!=-1;i=edge[i].next){        int v=edge[i].to;        if(v!=fa)            dfs(v,u);    }}void update_rev(int x){    if(!x) return;    swap(ch[x][0],ch[x][1]);    rev[x]^=1;}void push_down(int x){    if(rev[x]){        update_rev(ch[x][0]);        update_rev(ch[x][1]);        rev[x]=0;    }}void push_up(int x){    Max[x]=key[x];    Max[x]=max(Max[x],Max[ch[x][0]]);    Max[x]=max(Max[x],Max[ch[x][1]]);    sum[x]=key[x]+sum[ch[x][0]]+sum[ch[x][1]];}void rotate(int x){    int y=pre[x],d=ch[y][1]==x;    ch[y][d]=ch[x][!d];    pre[ch[y][d]]=y;    pre[x]=pre[y];    pre[y]=x;    ch[x][!d]=y;    if(rt[y]) rt[y]=false,rt[x]=true;    else ch[pre[x]][ch[pre[x]][1]==y]=x;    push_up(y);}//P函数先将根结点到x的路径上所有的结点的标记逐级下放void P(int x){    if(!rt[x]) P(pre[x]);    push_down(x);}//将x旋转到根void splay(int x){    P(x);    while(!rt[x]){        int f=pre[x],ff=pre[f];        if(rt[f]) rotate(x);        else if((ch[ff][1]==f)==(ch[f][1]==x))            rotate(f),rotate(x);        else            rotate(x),rotate(x);    }    push_up(x);}//将x到根的路径变成首选边int Access(int x){    int y=0;    for(;x;x=pre[y=x]){        splay(x);        rt[ch[x][1]]=true;        rt[ch[x][1]=y]=false;        push_up(x);    }    return y;}//判断是否是同根(真实的树，非splay)bool judge(int u,int v){    while(pre[u]) u=pre[u];    while(pre[v]) v=pre[v];    return u==v;}//让x成为它所在树的根void make_root(int x){    Access(x);    splay(x);    update_rev(x);}//调用后u是原来u和v的lca,v和ch[u][1]分别存着lca的2个儿子//(原来u和v所在的2颗子树)void lca(int &u,int &v){    Access(v),v=0;    while(u){        splay(u);        if(!pre[u]) return;        rt[ch[u][1]]=true;        rt[ch[u][1]=v]=false;        push_up(u);        u=pre[v=u];    }}//如果u,v不在同一棵子树中，则连接这2棵子树void link(int u,int v){    if(judge(u,v)){        printf("-1\n");        return;    }    make_root(u);    pre[u]=v;}//使u成为u所在树的根，并且将v和它父亲的边断开void cut(int u,int v){    if(u==v||!judge(u,v)){        printf("-1\n");        return;    }    make_root(u);    Access(v);    splay(v);    pre[ch[v][0]]=0;    rt[ch[v][0]]=true;    ch[v][0]=0;    push_up(v);}int main(){    int Q,u,v;    char str[20];    while(~scanf("%d",&n)){        init();        for(int i=1;i<n;i++){            scanf("%d%d",&u,&v);            addedge(u,v);            addedge(v,u);        }        for(int i=1;i<=n;i++)            scanf("%d",&key[i]);        dfs(1,0);        scanf("%d",&Q);        while(Q--){            scanf("%s%d%d",str,&u,&v);            if(strcmp(str,"CHANGE")==0){                Access(u);                splay(u);                key[u]=v;                push_up(u);            }            else if(strcmp(str,"QMAX")==0){                make_root(u);                Access(v);                splay(v);                printf("%d\n",Max[v]);            }            else{                make_root(u);                Access(v);                splay(v);                printf("%d\n",sum[v]);            }        }    }    return 0;}