BZOJ1036 [ZJOI2008]树的统计Count

本文版权归ljh2000和博客园共有，欢迎转载，但须保留此声明，并给出原文链接，谢谢合作。

 

 

本文作者：ljh2000
作者博客：http://www.cnblogs.com/ljh2000-jump/
转载请注明出处，侵权必究，保留最终解释权！

 

 

题目链接：BZOJ1036

 

正解：LCT

解题报告：

　　LCT裸题，连link和cut都不用了…

　　每次直接换根然后access一下，查询子树max和sum即可。

 

//It is made by ljh2000
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <ctime>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <string>
#include <complex>
using namespace std;
typedef long long LL;
typedef long double LB;
typedef complex<double> C;
const double pi = acos(-1);
const int MAXN = 30011;
const int MAXM = 60011;
int n,q,father[MAXN],tr[MAXN][2],mx[MAXN],sum[MAXN];
int tag[MAXN],top,stack[MAXN],a[MAXN],ecnt,first[MAXN],to[MAXM],next[MAXM];
char ch[12];
inline bool isroot(int x){ return (tr[father[x]][0]!=x) && (tr[father[x]][1]!=x); }
inline void upd(int x,int y){ if(mx[x]<mx[y]) mx[x]=mx[y];}
inline int getint(){
    int w=0,q=0; char c=getchar(); while((c<'0'||c>'9') && c!='-') c=getchar();
    if(c=='-') q=1,c=getchar(); while (c>='0'&&c<='9') w=w*10+c-'0',c=getchar(); return q?-w:w;
}

inline void update(int x){
	mx[x]=sum[x]=a[x]; int l=tr[x][0],r=tr[x][1];
	if(l) upd(x,l),sum[x]+=sum[l];
	if(r) upd(x,r),sum[x]+=sum[r];
}

inline void pushdown(int x){
	if(tag[x]==0) return ;
	int l=tr[x][0],r=tr[x][1];
	tag[x]=0; tag[l]^=1; tag[r]^=1;
	swap(tr[x][0],tr[x][1]);
}

inline void rotate(int x){
	int y,z; y=father[x]; z=father[y];
	int l,r; l=(tr[y][1]==x); r=l^1;
	if(!isroot(y)) tr[z][(tr[z][1]==y)]=x;
	father[x]=z; father[y]=x;
	tr[y][l]=tr[x][r]; father[tr[x][r]]=y; tr[x][r]=y;
	update(y); update(x);
}

inline void splay(int x){
	top=0; stack[++top]=x; int y,z;
	for(int i=x;!isroot(i);i=father[i]) stack[++top]=father[i];
	for(int i=top;i>=1;i--) pushdown(stack[i]);
	while(!isroot(x)) {
		y=father[x]; z=father[y];
		if(!isroot(y)) {
			if((tr[y][0]==x) ^ (tr[z][0]==y)) rotate(x);
			else rotate(y);
		}
		rotate(x);
	}
}

inline void access(int x){
	int last=0;
	while(x) {
		splay(x); tr[x][1]=last;
		update(x);//!!!
		last=x; x=father[x];
	}
}

inline void move_to_the_root(int x){
	access(x);
	splay(x);
	tag[x]^=1;
}

inline void dfs(int x,int fa){
	for(int i=first[x];i;i=next[i]) {
		int v=to[i]; if(v==fa) continue;
		father[v]=x;
		dfs(v,x);
	}
}

inline void work(){
	n=getint(); int x,y;
	for(int i=1;i<n;i++) {
		x=getint(); y=getint();
		next[++ecnt]=first[x]; first[x]=ecnt; to[ecnt]=y;
		next[++ecnt]=first[y]; first[y]=ecnt; to[ecnt]=x;
	}
	for(int i=1;i<=n;i++) a[i]=getint(),mx[i]=sum[i]=a[i];
	dfs(1,0);
	q=getint();
	while(q--) {
		scanf("%s",ch); x=getint(); y=getint();
		if(ch[0]=='C') {
			splay(x);
			a[x]=y;
			update(x);
		}
		else if(ch[1]=='S') {
			move_to_the_root(x);
			access(y); splay(y);
			printf("%d\n",sum[y]);
		}
		else {
			move_to_the_root(x);
			access(y); splay(y);
			printf("%d\n",mx[y]);
		}
	}
}

int main()
{
    work();
    return 0;
}