本文主要是介绍poj3237--Tree(树链剖分+线段树)，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
题目链接：poj--3237
题意很简单，给出n个节点的一棵树，有三种操作：
1、C修改第i条边的值为v
2、N改变节点a到b内边的权值的符号（取反）
3、Q询问节点a到b内权值的最大值
首先树链剖分，将边整合到线段树上，线段树数组cl，因为存在取反操作，所以最大值可能是由最小值取反得到，所以记录最大和最小值，cl[i][0]记录第i段的最大值，cl[i][1]记录最小值，lazy做标记，该段是否取反，两次取反==没有取反
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;
#define LL __int64
#define INF 0x3f3f3f3f
#define maxn 11000
#define int_now int l,int r,int rt
#define now l,r,rt
#define lson l,(l+r)/2,rt<<1
#define rson (l+r)/2+1,r,rt<<1|1
struct node{
int u , v , w ;
int next ;
} edge[maxn<<1] ;
int head[maxn] , cnt , vis[maxn] ;
int num[maxn] , dep[maxn] , fa[maxn] , son[maxn] , top[maxn] , w[maxn] , step ;
int cl[maxn<<2][2] , lazy[maxn<<2] , n ;//cl[i][0]最大值,cl[i][1]最小值
void add(int u,int v,int w) {
edge[cnt].u = u ; edge[cnt].v = v ; edge[cnt].w = w ;
edge[cnt].next = head[u] ; head[u] = cnt++ ;
return ;
}
void dfs1(int u) {
int i , v ;
num[u] = 1 ;
son[u] = -1 ;
for(i = head[u] ; i != -1 ; i = edge[i].next) {
v = edge[i].v ;
if( vis[v] ) continue ;
vis[v] = 1 ;
dep[v] = dep[u] + 1 ;
fa[v] = u ;
dfs1(v) ;
num[u] += num[v] ;
if( son[u] == -1 || ( num[son[u]] < num[v] ) )
son[u] = v ;
}
return ;
}
void dfs2(int u) {
if( son[u] == -1 ) return ;
top[son[u]] = top[u] ;
w[ son[u] ] = step++ ;
vis[ son[u] ] = 1 ;
dfs2(son[u]) ;
int i , v ;
for(i = head[u] ; i != -1 ; i = edge[i].next) {
v = edge[i].v ;
if( vis[v] ) continue ;
vis[v] = 1 ;
top[v] = v ;
w[v] = step++ ;
dfs2(v) ;
}
return ;
}
void dfs() {
memset(vis,0,sizeof(vis)) ;
vis[1] = 1 ; dep[1] = 1 ; fa[1] = -1 ;
dfs1(1) ;
memset(vis,0,sizeof(vis)) ;
vis[1] = 1 ; top[1] = 1 ; step = 1 ;
dfs2(1) ;
return ;
}
void swap1(int rt) {
cl[rt][0] = -cl[rt][0] ;
cl[rt][1] = -cl[rt][1] ;
swap(cl[rt][0],cl[rt][1]) ;
}
void push_up(int_now) {
cl[rt][0] = max(cl[rt<<1][0],cl[rt<<1|1][0]) ;
cl[rt][1] = min(cl[rt<<1][1],cl[rt<<1|1][1]) ;
if( lazy[rt] )
swap1(rt) ;
return ;
}
void push_down(int_now) {
if( lazy[rt] ) {
lazy[rt] = 0 ;
lazy[rt<<1] = 1 - lazy[rt<<1] ;
lazy[rt<<1|1] = 1 - lazy[rt<<1|1] ;
swap1(rt<<1) ;
swap1(rt<<1|1) ;
}
return ;
}
void update1(int k,int x,int_now) {
if( l == k && r == k ) {
lazy[rt] = 0 ;
cl[rt][0] = cl[rt][1] = x ;
return ;
}
push_down(now) ;
if( k <= (l+r)/2 )
update1(k,x,lson) ;
else
update1(k,x,rson) ;
push_up(now) ;
return ;
}
void update2(int ll,int rr,int_now) {
if( ll > r || rr < l ) return ;
if( ll <= l && rr >= r ) {
lazy[rt] = 1 - lazy[rt] ;
swap1(rt) ;
return ;
}
push_down(now) ;
update2(ll,rr,lson) ;
update2(ll,rr,rson) ;
push_up(now) ;
return ;
}
int query(int ll,int rr,int_now,int sum) {
if( ll > r || rr < l ) return -INF ;
if( ll <= l && rr >= r ) {
if( sum%2 ) return -cl[rt][1] ;
else return cl[rt][0] ;
}
sum += lazy[rt] ;
return max( query(ll,rr,lson,sum),query(ll,rr,rson,sum) ) ;
}
void solve(int u,int v,int k) {
int f1 , f2 , ans = -INF ;
while( u != v ) {
if( dep[u] > dep[v] )
swap(u,v) ;
f1 = top[u] ; f2 = top[v] ;
if( f1 == f2 ) {
if( k )
ans = max(ans,query(w[son[u]],w[v],1,step-1,1,0)) ;
else
update2(w[son[u]],w[v],1,step-1,1) ;
v = u ;
}
else if( dep[f1] > dep[f2] ) {
if( k )
ans = max(ans,query(w[f1],w[u],1,step-1,1,0)) ;
else
update2(w[f1],w[u],1,step-1,1) ;
u = fa[f1] ;
}
else{
if( k )
ans = max(ans,query(w[f2],w[v],1,step-1,1,0)) ;
else
update2(w[f2],w[v],1,step-1,1);
v = fa[f2] ;
}
}
if( ans == -INF ) ans = 0 ;
if( k )
printf("%d\n", ans) ;
return ;
}
char str[100] ;
int main() {
int t , i , j , k ;
int u , v , s ;
scanf("%d", &t) ;
while( t-- ) {
memset(head,-1,sizeof(head)) ;
cnt = 0 ;
scanf("%d", &n) ;
for(i = 1 ; i < n ; i++) {
scanf("%d %d %d", &u, &v, &s) ;
add(u,v,s) ;
add(v,u,s) ;
}
dfs() ;
memset(cl,0,sizeof(cl)) ;
memset(lazy,0,sizeof(lazy)) ;
for(i = 0 ; i < n-1 ; i++) {
if( dep[ edge[i*2].u ] > dep[ edge[i*2].v ] )
swap(edge[i*2].u,edge[i*2].v ) ;
update1(w[ edge[i*2].v ],edge[i*2].w,1,step-1,1) ;
}
while( scanf("%s", str) ) {
if( str[0] == 'D' ) break ;
if( str[0] == 'C' ) {
scanf("%d %d", &i, &k) ;
update1(w[ edge[(i-1)*2].v ],k,1,step-1,1) ;
}
else if( str[0] == 'N' ) {
scanf("%d %d", &i, &j) ;
solve(i,j,0) ;
}
else{
scanf("%d %d", &i, &j) ;
solve(i,j,1) ;
}
}
}
return 0 ;
}
这篇关于poj3237--Tree(树链剖分+线段树)的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
