bzoj 1036（树链剖分+线段树）

继续练一下链剖吧。。这次写的蛮久的不过还是尽量避免了盲目使用gdb

看到单点修改其实蛮想用BIT的不过鉴于要求区间最大值BIT还是用不上啊，用log^2n的算法担心会超时。。所以这题应该算是比较标准的链剖了。。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
#include<cmath>
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,l,r) for(int i=l;i>=r;i--)
#define link(x) for(edge *j=h[x];j;j=j->next)
#define inf 1000000007
#define eps 1e-8
#define mem(a) memset(a,0,sizeof(a))
#define ll long long
#define succ(x) (1<<x)
#define lowbit(x) (x&(-x))
#define sqr(x) ((x)*(x))
#define ls T[i<<1]
#define rs T[i<<1|1]
#define op T[i]
#define NM 30005
#define nm 60005
#define pi 3.141592653
using namespace std;
int read(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return f*x;
}

struct edge{int t;edge*next;}e[nm],*h[NM],*o=e;
void add(int x,int y){o->t=y;o->next=h[x];h[x]=o++;}
int top[NM],TOP,size[NM],son[NM],d[NM],f[NM],tot,id[NM];
int n,m,_x,_y,a[NM];
char _s[10];
void dfs1(int x){
	link(x)if(!f[j->t]){
		f[j->t]=x;d[j->t]=d[x]+1;
		dfs1(j->t);
		if(size[j->t]>size[son[x]])son[x]=j->t;
		size[x]+=size[j->t];
	}
	size[x]++;
}
void dfs2(int x){
	top[x]=TOP;id[x]=++tot;
	if(son[x])dfs2(son[x]);
	link(x)if(!top[j->t])dfs2(TOP=j->t);
}

struct info{
	ll s;int m,size;
	info operator+(const info&o){
		info f;f.s=s+o.s;
		f.size=size+o.size;
		f.m=max(m,o.m);
		return f;
	}
}T[4*NM];

void build(int i,int x,int y){
	int t=x+y>>1;
	if(x==y){op.s=op.m=a[x];op.size=1;return;}
	build(i<<1,x,t);build(i<<1|1,t+1,y);
	op=rs+ls;
}
void mod(int i,int x,int y){
	int t=x+y>>1;
	if(x==y){op.s=op.m=_y;return;}
	if(_x<=t)mod(i<<1,x,t);else mod(i<<1|1,t+1,y);
	op=ls+rs;
}
ll sum(int i,int x,int y){
	int t=x+y>>1;
	if(_y<x||y<_x)return 0;
	if(_x<=x&&y<=_y)return op.s;
	return sum(i<<1,x,t)+sum(i<<1|1,t+1,y);
}
int query(int i,int x,int y){
	int t=x+y>>1;
	if(_y<x||y<_x)return -inf;
	if(_x<=x&&y<=_y)return op.m;
	return max(query(i<<1,x,t),query(i<<1|1,t+1,y));
}

int  _query1(int x,int y){
	int s=-inf;
	for(;top[x]!=top[y];x=f[top[x]]){
		if(d[top[x]]<d[top[y]])swap(x,y);
		_x=id[top[x]];_y=id[x];s=max(s,query(1,1,n));
	}
	if(d[x]>d[y])swap(x,y);
	_x=id[x];_y=id[y];s=max(s,query(1,1,n));
	return s;
}

ll _query2(int x,int y){
	ll s=0;
	for(;top[x]!=top[y];x=f[top[x]]){
		if(d[top[x]]<d[top[y]])swap(x,y);
		_x=id[top[x]];_y=id[x];s+=sum(1,1,n);
	}
	if(d[x]>d[y])swap(x,y);
	_x=id[x];_y=id[y];s+=sum(1,1,n);
	return s;
}

int main(){
	freopen("data.in","r",stdin);
	n=read();
	inc(i,1,n-1){_x=read();_y=read();add(_x,_y);add(_y,_x);}
	dfs1(f[1]=1);dfs2(TOP=1);
	inc(i,1,n)a[id[i]]=read();
	build(1,1,n);
	m=read();
	while(m--){
		scanf("%s",_s);
		_x=read();_y=read();
		if(_s[0]=='C'){
			_x=id[_x];mod(1,1,n);
		}else if(_s[1]=='M')printf("%d\n",_query1(_x,_y));
		else printf("%lld\n",_query2(_x,_y));
	}
	return 0;
}

1036: [ZJOI2008]树的统计Count

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

Source

[ Submit][ Status][ Discuss]