[bzoj1984]月下“毛景树” 树链剖分

1984: 月下“毛景树”

Time Limit: 20 Sec   Memory Limit: 64 MB
[ Submit][ Status][ Discuss]

Description

毛毛虫经过及时的变形，最终逃过的一劫，离开了菜妈的菜园。 毛毛虫经过千山万水，历尽千辛万苦，最后来到了小小的绍兴一中的校园里。爬啊爬~爬啊爬~~毛毛虫爬到了一颗小小的“毛景树”下面，发现树上长着他最爱吃的毛毛果~~~ “毛景树”上有N个节点和N-1条树枝，但节点上是没有毛毛果的，毛毛果都是长在树枝上的。但是这棵“毛景树”有着神奇的魔力，他能改变树枝上毛毛果的个数：  Change k w：将第k条树枝上毛毛果的个数改变为w个。  Cover u v w：将节点u与节点v之间的树枝上毛毛果的个数都改变为w个。  Add u v w：将节点u与节点v之间的树枝上毛毛果的个数都增加w个。 由于毛毛虫很贪，于是他会有如下询问：  Max u v：询问节点u与节点v之间树枝上毛毛果个数最多有多少个。

Input

第一行一个正整数N。 接下来N-1行，每行三个正整数Ui，Vi和Wi，第i+1行描述第i条树枝。表示第i条树枝连接节点Ui和节点Vi，树枝上有Wi个毛毛果。 接下来是操作和询问，以“Stop”结束。

Output

对于毛毛虫的每个询问操作，输出一个答案。

Sample Input

4
1 2 8
1 3 7
3 4 9
Max 2 4
Cover 2 4 5
Add 1 4 10
Change 1 16
Max 2 4
Stop

Sample Output

9
16

【Data Range】
1<=N<=100,000，操作+询问数目不超过100,000。
保证在任意时刻，所有树枝上毛毛果的个数都不会超过10^9个。

HINT

Source

change应该比modify优先级高

这道题我没有压到100行以内

#include<iostream>
#include<cstring>
#include<cstdio>
#define inf 0x7fffffff
const int N = 100000 + 5;
using namespace std;
int cnt=1,n,sz; char ch[10];
int pos[N],bel[N],last[N],dep[N],siz[N],anc[N][20],id[N];
int ls[N<<2],rs[N<<2],mx[N<<2],l[N<<2],r[N<<2],mf[N<<2],af[N<<2],root,idd;
struct Edge{int to,next,v;}e[N<<1];
void insert(int u,int v,int w){
	e[++cnt].to = v; e[cnt].next = last[u]; e[cnt].v = w; last[u] = cnt;
	e[++cnt].to = u; e[cnt].next = last[v]; e[cnt].v = w; last[v] = cnt;
}
void build( int &k, int L, int R ){
	k = ++idd; l[idd] = L; r[idd] = R; mf[k] = -1;
	if( L == R ) return; int mid = (L+R)>>1;
	build( ls[k], L, mid ); build( rs[k], mid+1, R );
}
void update( int k ){
	mx[k] = max(mx[ls[k]],mx[rs[k]]);
}
void pushdown( int k ){
	if( l[k] == r[k] ) return;
	if( mf[k] != -1 ){
		af[ls[k]] = af[rs[k]] = 0;
		mx[ls[k]] = mx[rs[k]] = mf[ls[k]] = mf[rs[k]] = mf[k];
		mf[k] = -1;
	} if( af[k] ){
		mx[ls[k]] += af[k]; mx[rs[k]] += af[k];
		if( mf[ls[k]] != -1 ) mf[ls[k]] += af[k]; else af[ls[k]] += af[k];
		if( mf[rs[k]] != -1 ) mf[rs[k]] += af[k]; else af[rs[k]] += af[k];
		af[k] = 0;
	}
}
void change( int k, int x, int y, int v ){
	pushdown(k);
	if( x == l[k] && y == r[k] ){ mx[k] = mf[k] = v; return;}
	int mid = (l[k]+r[k])>>1;
	if( mid >= y ) change(ls[k],x,y,v);
	else if( mid < x ) change(rs[k],x,y,v);
	else {
		change(ls[k],x,mid,v);
		change(rs[k],mid+1,y,v);
	}
	update(k);
}
void modify( int k, int x, int y, int v ){
	pushdown(k);
	if( x == l[k] && y == r[k] ){ mx[k] += v; af[k] = v; return;}
	int mid = (l[k]+r[k])>>1;
	if( mid >= y ) modify(ls[k],x,y,v);
	else if( mid < x ) modify(rs[k],x,y,v);
	else {
		modify(ls[k],x,mid,v);
		modify(rs[k],mid+1,y,v);
	}
	update(k);
}
int query( int k, int x, int y ){
	pushdown(k);
	if( l[k] == x && r[k] == y ) return mx[k];
	int mid = (l[k]+r[k])>>1;
	if( mid >= y ) return query(ls[k],x,y);
	else if( mid < x ) return query(rs[k],x,y);
	else return max(query(ls[k],x,mid),query(rs[k],mid+1,y));
}
void solve_change( int x, int f, int v ){
	while( bel[x] != bel[f] ){
		change(root,pos[bel[x]],pos[x],v);
		x = anc[bel[x]][0];
	}
	if( pos[f]+1 <= pos[x] ) change(root,pos[f]+1,pos[x],v);
}
void solve_modify( int x, int f, int v ){
	while( bel[x] != bel[f] ){
		modify(root,pos[bel[x]],pos[x],v);
		x = anc[bel[x]][0];
	}
	if( pos[f]+1 <= pos[x] ) modify(root,pos[f]+1,pos[x],v);
}
int solve_query( int x, int f ){
	int res = -inf;
	while( bel[x] != bel[f] ){
		res = max(res,query(root,pos[bel[x]],pos[x]));
		x = anc[bel[x]][0];
	}
	if( pos[f]+1 <= pos[x] ) res = max(res,query(root,pos[f]+1,pos[x]));
	return res;
}
void dfs1( int x, int f ){
	siz[x] = 1;
	for( int p = 1; p <= 16; p++ )
		if( (1<<p) <= dep[x] ) anc[x][p] = anc[anc[x][p-1]][p-1];
		else break;
	for( int i = last[x]; i; i = e[i].next )
		if( e[i].to != f ){
			dep[e[i].to] = dep[x]+1;
			anc[e[i].to][0] = x;
			dfs1(e[i].to,x);
			siz[x] += siz[e[i].to];
		}
}
void dfs2( int x, int chain ){
	int k = 0; bel[x] = chain; pos[x] = ++sz;
	for( int i = last[x]; i; i = e[i].next )
		if( dep[e[i].to] > dep[x] ){
			if( siz[e[i].to] > siz[k] ) k = e[i].to;
		} else{ id[i>>1] = x; modify(1,pos[x],pos[x],e[i].v); }
	if( !k ) return; dfs2(k,chain);
	for( int i = last[x]; i; i = e[i].next )
		if( dep[e[i].to] > dep[x] && e[i].to != k ) dfs2(e[i].to,e[i].to);
}
int lca( int x, int y ){
	if( dep[x] < dep[y] ) swap(x,y);
	int t = dep[x]-dep[y];
	for( int p = 0; p <= 16; p++ ) if( (1<<p)&t ) x = anc[x][p];
	if( x == y ) return x;
	for( int p = 16; p >= 0; p-- ) if( anc[x][p] != anc[y][p] ) x = anc[x][p], y = anc[y][p];
	return anc[x][0];
}
int main(){
	scanf("%d", &n);
	for( int i = 1, u, v, w; i < n; i++ ) scanf("%d%d%d", &u, &v, &w), insert( u, v, w );
	build( root, 1, n ); dfs1(1,0); dfs2(1,1);
	while( 1 ){
		int u,v,w,f;
		scanf("%s", ch);
		switch(ch[1]){
			case 't':return 0;break;
			case 'd':scanf("%d%d%d",&u,&v,&w);f=lca(u,v); solve_modify(u,f,w);solve_modify(v,f,w);break;
			case 'o':scanf("%d%d%d",&u,&v,&w);f=lca(u,v); solve_change(u,f,w);solve_change(v,f,w);break;
			case 'h':scanf("%d%d",&u,&w);change(1,pos[id[u]],pos[id[u]],w);break;
			case 'a':scanf("%d%d",&u,&v);f=lca(u,v);printf("%d\n",max(solve_query(u,f),solve_query(v,f)));break;
		}
	}
}