题目链接:POJ 3237 Tree
题意:在树上有三种操作。
CHANGE A B 1.将第A条边的值赋值为B;
NEGATE A B 2.将A节点到B节点路径上的边变为其相反数。
QUERY A B 3.访问A节点到B节点的路径上的边权最大。
比较明显的树链剖分,主要是在线段树区间更新,取相反数可以得到,最大值和最小值取相反数后,再交换就是各自改变后的值。
AC代码:
#include <string.h>
#include <stdio.h>
#include <algorithm>
#define IN freopen ("in.txt" , "r" , stdin);
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int maxn=100010+10;
const int inf=100000000;
struct Edge{
int to,next;
};
struct Edge edge[maxn*2];
int head[maxn],tot;
int stree[maxn];//节点v在线段树上的编号
int pre[maxn];//在线段树中编号为v节点所对应原图中的编号。
int fa[maxn];//f[v] 表示v的父亲节点
int son[maxn];//表示 与v同在一重链上的儿子节点(重儿子)
int num[maxn];//num[v] v为根的子树的节点数。
int top[maxn];//top[v] v所在链的顶端节点。
int deep[maxn];//deep[v] 表示v的深度(根的深度为1)
int data[maxn];
int pos,n;
void addedge(int u,int v){
edge[tot].to=v;
edge[tot].next=head[u];
head[u]=tot++;
}
void dfs1(int u,int father,int dep){
int i;
deep[u]=dep;
fa[u]=father;
num[u]=1;
for(i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].to;
if(v!=father){
dfs1(v,u,dep+1);
num[u]+=num[v];
if(son[u]==-1 || num[v]>num[son[u]]){
son[u]=v;
}
}
}
}
//给点编号得到线段树上的对应标号
void getpos(int u,int sp){
int i;
top[u]=sp;
stree[u]=pos++;
pre[stree[u]]=u;
if(son[u]==-1)
return ;
getpos(son[u],sp);
for(i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].to;
if(v!=son[u] && v!=fa[u]){
getpos(v,v);
}
}
}
void init(){
tot=0;
memset(head,-1,sizeof head);
pos=0;
memset(son,-1,sizeof son);
}
//树状数组or线段树or其他
struct node{
int l,r;
int min,max;
int laz;
};
struct node tree[maxn*3];
void PushUp(int rt){
tree[rt].max=max(tree[rt<<1].max,tree[rt<<1|1].max);
tree[rt].min=min(tree[rt<<1].min,tree[rt<<1|1].min);
}
void PushDown(int rt){
int t;
if(tree[rt].l==tree[rt].r) return ;
if(tree[rt].laz){
tree[rt<<1].laz+=tree[rt].laz;
tree[rt<<1].laz%=2;
tree[rt<<1|1].laz+=tree[rt].laz;
tree[rt<<1|1].laz%=2;
tree[rt].laz=0;
t=tree[rt<<1].max;
tree[rt<<1].max=-tree[rt<<1].min;
tree[rt<<1].min=-t;
t=tree[rt<<1|1].max;
tree[rt<<1|1].max=-tree[rt<<1|1].min;
tree[rt<<1|1].min=-t;
}
}
void build(int l,int r,int rt){
tree[rt].l=l;
tree[rt].r=r;
tree[rt].max=0;
tree[rt].min=0;
tree[rt].laz=0;
if(l==r){
return ;
}
int m=(tree[rt].l+tree[rt].r)>>1;
build(lson);
build(rson);
PushUp(rt);
}
void update(int p,int add,int rt){
if(tree[rt].l==tree[rt].r && tree[rt].l==p){
tree[rt].max=add;
tree[rt].min=add;
tree[rt].laz=0;
return ;
}
PushDown(rt);
int m=(tree[rt].l+tree[rt].r)>>1;
if(p<=m) update(p,add,rt<<1);
else update(p,add,rt<<1|1);
PushUp(rt);
}
void update2(int L,int R,int rt){
if(L<=tree[rt].l && tree[rt].r<=R){
int t=tree[rt].max;
tree[rt].max=-tree[rt].min;
tree[rt].min=-t;
tree[rt].laz+=1;
tree[rt].laz%=2;
return ;
}
PushDown(rt);
int m=(tree[rt].l+tree[rt].r)>>1;
if(L<=m) update2(L,R,rt<<1);
if(R>m) update2(L,R,rt<<1|1);
PushUp(rt);
}
int query_max(int L,int R,int rt){
if(L<=tree[rt].l && tree[rt].r<=R){
return tree[rt].max;
}
PushDown(rt);
int m=(tree[rt].l+tree[rt].r)>>1;
int ret=-inf;
if(L<=m) ret=max(ret,query_max(L,R,rt<<1));
if(R>m) ret=max(ret,query_max(L,R,rt<<1|1));
PushUp(rt);
return ret;
}
int find_max(int u,int v){
int f1,f2,ans=-inf;
f1=top[u];
f2=top[v];
while(f1!=f2){
//始终f1的深度比f2要大。深度大先更新
if(deep[f1]<deep[f2]){
swap(f1,f2);
swap(u,v);
}
ans=max(ans,query_max(stree[f1],stree[u],1));
u=fa[f1];
f1=top[u];
}
if(u==v) return ans;
if(deep[u]>deep[v])
swap(u,v);
ans=max(ans,query_max(stree[son[u]],stree[v],1));
return ans;
}
void change(int u,int v){
int f1,f2;
f1=top[u];
f2=top[v];
while(f1!=f2){
//始终f1的深度比f2要大。深度大先更新
if(deep[f1]<deep[f2]){
swap(f1,f2);
swap(u,v);
}
update2(stree[f1],stree[u],1);
u=fa[f1];
f1=top[u];
}
if(u==v) return ;
if(deep[u]>deep[v])
swap(u,v);
update2(stree[son[u]],stree[v],1);
return ;
}
int e[maxn][5];
int main()
{
int t,x,y;
int q,i;
scanf("%d",&t);
while(t--){
init();
scanf("%d",&n);
for(i=0;i<n-1;i++){
scanf("%d %d %d",&e[i][0],&e[i][1],&e[i][2]);
addedge(e[i][0],e[i][1]);
addedge(e[i][1],e[i][0]);
}
dfs1(1,0,0);//根的deep是0.father从0开始。
getpos(1,1);
build(0,pos-1,1);
for(i=0;i<n-1;i++){
if(deep[e[i][0]]>deep[e[i][1]])
swap(e[i][0],e[i][1]);
update(stree[e[i][1]],e[i][2],1);
}
char op[10];
while(1){
int a,b;
scanf("%s",op);
if(op[0]=='D')
break;
scanf("%d %d",&a,&b);
if(op[0]=='Q')
printf("%d\n",find_max(a,b));
else if(op[0]=='C')
update(stree[e[a-1][1]],b,1);
else
change(a,b);
}
}
return 0;
}
/*
10
7
1 2 1
1 3 2
3 4 3
3 5 4
5 6 5
5 7 6
NEGATE 1 3
QUERY 1 4
NEGATE 2 4
CHANGE 4 40
QUERY 4 5
NEGATE 2 7
QUERY 2 7
QUERY 3 5
CHANGE 4 30
NEGATE 2 7
QUERY 4 7
QUERY 4 5
QUERY 2 7
DONE
10
6
1 2 1
2 5 2
2 6 3
1 3 4
3 4 5
QUERY 4 5
CHANGE 1 3
QUERY 4 5
*/