BZOJ 1036 [ZJOI2008]树的统计Count

1036: [ZJOI2008]树的统计Count

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 8095  Solved: 3300
[Submit][Status][Discuss]

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会挂，就不犯懒了吧。。。

我擦。。。树剖还写错了一发。。。手残错。。。。

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cmath>
 4 #include<algorithm>
 5 #include<queue>
 6 #include<cstring>
 7 #define PAU putchar(' ')
 8 #define ENT putchar('\n')
 9 #define CH for(int d=0;d<2;d++)if(ch[d])
10 #define lson x->ch[0],L,M
11 #define rson x->ch[1],M+1,R
12 using namespace std;
13 const int maxn=30000+10,maxnode=60000+10,inf=-1u>>1;
14 struct node{
15     node*ch[2];int mx,sm,siz;node(){mx=-inf;sm=0;}
16     void init(int a){mx=sm=a;return;}
17     void sett(int a){mx=a;sm=siz*a;return;}
18     void update(){sm=0;mx=-inf;CH{sm+=ch[d]->sm;mx=max(mx,ch[d]->mx);}return;}
19 }seg[maxnode],*nodecnt=seg,*root;
20 int A[maxn],n,Q,pos,cv,_mx,_sm,ql,qr;
21 void build(node*&x=root,int L=1,int R=n){
22     x=nodecnt++;int M=L+R>>1;if(L==R)x->init(A[M]);
23     else build(lson),build(rson),x->update();x->siz=R-L+1;return;
24 }
25 void update(node*&x=root,int L=1,int R=n){
26     if(L==R){x->sett(cv);return;}
27     int M=L+R>>1;if(pos<=M)update(lson);else update(rson);x->update();return;
28 }
29 void query(node*x=root,int L=1,int R=n){
30     if(ql<=L&&R<=qr){
31         _mx=max(_mx,x->mx);
32         _sm+=x->sm;
33     }else{int M=L+R>>1;
34         if(ql<=M)query(lson);if(qr>M)query(rson);
35     }return;
36 }
37 struct ted{int x,y;ted*nxt;}adj[maxn<<1],*fch[maxn],*ms=adj;
38 void add(int x,int y){
39     *ms=(ted){x,y,fch[x]};fch[x]=ms++;*ms=(ted){y,x,fch[y]};fch[y]=ms++;return;
40 }
41 int top[maxn],p[maxn],sz=0,siz[maxn],son[maxn],fa[maxn],dep[maxn];
42 void dfs(int x){
43     siz[x]=1;dep[x]=dep[fa[x]]+1;
44     for(ted*e=fch[x];e;e=e->nxt){
45         int v=e->y;if(v!=fa[x]){
46             fa[v]=x;dfs(v);siz[x]+=siz[v];
47             if(siz[son[x]]<siz[v])son[x]=v;
48         }
49     }return;
50 }
51 void build(int x,int tp){
52     p[x]=++sz;top[x]=tp;
53     if(son[x])build(son[x],tp);
54     for(ted*e=fch[x];e;e=e->nxt){
55         int v=e->y;if(v!=fa[x]&&v!=son[x])build(v,v);
56     }return;
57 }
58 void query(int x,int y){
59     int f1=top[x],f2=top[y];
60     while(f1!=f2){
61         if(dep[f1]<dep[f2])swap(f1,f2),swap(x,y);
62         ql=p[f1];qr=p[x];query();x=fa[f1];f1=top[x];
63     }if(dep[x]>dep[y])swap(x,y);ql=p[x];qr=p[y];query();return;
64 }
65 inline int read(){
66     int x=0,sig=1;char ch=getchar();
67     for(;!isdigit(ch);ch=getchar())if(ch=='-')sig=0;
68     for(;isdigit(ch);ch=getchar())x=10*x+ch-'0';
69     return sig?x:-x;
70 }
71 inline void write(int x){
72     if(x==0){putchar('0');return;}if(x<0)putchar('-'),x=-x;
73     int len=0,buf[15];while(x)buf[len++]=x%10,x/=10;
74     for(int i=len-1;i>=0;i--)putchar(buf[i]+'0');return;
75 }
76 void init(){
77     n=read();
78     for(int i=1;i<n;i++)add(read(),read());
79     dfs(1);build(1,1);
80     for(int i=1;i<=n;i++)A[p[i]]=read();
81     build();
82     return;
83 }
84 void work(){
85     Q=read();int x,y;char s[10];
86     while(Q--){
87         scanf("%s",s);
88         if(s[0]=='C')pos=p[read()],cv=read(),update();
89         else{_mx=-inf;_sm=0;query(read(),read());
90             if(s[1]=='M')write(_mx),ENT;else write(_sm),ENT;
91         }
92     }
93     return;
94 }
95 void print(){
96     return;
97 }
98 int main(){init();work();print();return 0;}