[bzoj3306]树 dfs序+线段树

3306: 树

Time Limit: 10 Sec   Memory Limit: 256 MB
[ Submit][ Status][ Discuss]

Description

给定一棵大小为 n 的有根点权树,支持以下操作: 
　　• 换根 
　　• 修改点权  
    　• 查询子树最小值 

Input

　　第一行两个整数 n, Q ,分别表示树的大小和操作数。 
　　接下来n行,每行两个整数f,v,第i+1行的两个数表示点i的父亲和点i的权。保证f < i。如 果f = 0,那么i为根。输入数据保证只有i = 1时,f = 0。 
　　接下来 m 行,为以下格式中的一种: 
　　• V x y表示把点x的权改为y 
　　• E x 表示把有根树的根改为点 x 
　　• Q x 表示查询点 x 的子树最小值 

Output

　　对于每个 Q ,输出子树最小值。 

Sample Input

3 7
0 1
1 2
1 3
Q 1
V 1 6
Q 1
V 2 5
Q 1
V 3 4
Q 1

Sample Output

1
2
3
4

HINT

　　对于 100% 的数据:n, Q ≤ 10^5。

Source

换根已经搞死我了

#include<bits/stdc++.h>
using namespace std;
const int N = 1000000 + 5;
int last[N],cnt,q,ans,tot=0,n,root=1,in[N],out[N],dep[N],p[N],flag[N],sum[N],w[N],id,anc[N][20];
struct Edge{ int to,next; }e[N*2];
void insert( int u, int v ){ e[++cnt].to = v; e[cnt].next = last[u]; last[u] = cnt; }
void dfs( int x, int f ){
	in[x] = ++tot; dep[x] = dep[f] + 1; anc[x][0] = f;
	for( int i = 1; i <= 19; i++ ) anc[x][i] = anc[anc[x][i-1]][i-1];
	for( int i = last[x]; i; i = e[i].next )
		if( e[i].to ^ f )
			dfs( e[i].to, x );
	out[x] = tot;
}
void pushdown( int k, int l, int r ){
	if( flag[k] ){
		sum[k<<1] = flag[k]; flag[k<<1] = flag[k];
		sum[k<<1|1] = flag[k]; flag[k<<1|1] = flag[k];
		flag[k] = 0;
	}
}
void update( int k ){
	sum[k] = min( sum[k<<1], sum[k<<1|1] );
}
void modify( int k, int l, int r, int L, int R, int val ){
	if( l >= L && r <= R ){
		flag[k] = val; sum[k] = val; return;
	}
	int mid = l + r >> 1;
	pushdown( k, l, r );
	if( L <= mid ) modify( k<<1, l, mid, L, R, val );
	if( R >  mid ) modify( k<<1|1, mid+1, r, L, R, val );
	update( k );
}
int query( int k, int l, int r, int L, int R ){
	if( l >= L && R >= r ){
		return sum[k];
	}
	pushdown( k, l, r );
	int mid = l + r >> 1, res = 2147483647;
	if( L <= mid ) res = min( res, query( k<<1, l, mid, L, R ) );
	if( R >  mid ) res = min( res, query( k<<1|1, mid+1, r, L, R ) );
	return res;
}
int up( int x, int p ){
	for( int i = 19; i >= 0; i-- )
		if( (1<<i) & p ) x = anc[x][i];
	return x;
}
int main(){
	cin>>n>>q;
	for( int i = 1,u,v; i <= n; i++ ){
		scanf("%d%d", &u, &v); if( u == 0 ) id = i;
		else insert( u, i ), insert( i, u );
		w[i] = v;
	}
	dep[id] = 1; dfs( id, 0 );
	for( int i = 1; i <= n; i++ ) modify( root, 1, n, in[i], in[i], w[i] );
	for( int i = 1; i <= q; i++ ){
		char opt[5]; int x, y;
		scanf("%s", opt);
		if( opt[0] == 'V' ){
			scanf("%d%d", &x, &y);
			modify( root, 1, n, in[x], in[x], y );
		}if( opt[0] == 'E' ){
			scanf("%d", &id);
		}if( opt[0] == 'Q' ){
			scanf("%d", &x);
			if( x == id ) printf("%d\n", query( root, 1, n, 1, n ) );
			else if( in[x] <= in[id] && out[id] <= out[x] ){
				int y = up( id, dep[id]-dep[x]-1 );
				printf("%d\n", min( query( root, 1, n, 1, in[y]-1 ), query( root, 1, n, out[y]+1, n) ) );
			}else printf("%d\n", query( root, 1, n, in[x], out[x] ) );
		}
	}
	return 0;
}