poj 3237(树链剖分+线段树)

题意：给一棵树，三种操作。将第i条边的权值改为v，将a到b的路径上的边的权值全部取反，求a到b路径上边的权值的最大值。

思路：明显的树链剖分，加上线段树的操作。因为有取反的操作所以每个区间要记录最大值和最小值。查询两点间的路径时，用求公共祖先的方式去求。

#include<iostream>
#include<stdio.h>
#include<string.h>
const int N=101000;
const int inf=0x3fffffff;
using namespace std;
int head
,num,son
,sz
,father
,dep
,idx,a
,cot
,ti
,top
;
struct edge
{
int st,ed,w,next;
}e[N*4];
void addedge(int x,int y,int w)
{
e[num].st=x;e[num].ed=y;e[num].w=w;e[num].next=head[x];head[x]=num++;
e[num].st=y;e[num].ed=x;e[num].w=w;e[num].next=head[y];head[y]=num++;
}
int max(int a,int b)
{
if(a>b)return a;
return b;
}
int min(int a,int b)
{
if(a>b)return b;
return a;
}
//******************树链剖分****************************
void find_son(int u,int fa)
{
int i,v;
son[u]=0;sz[u]=1;
for(i=head[u];i!=-1;i=e[i].next)
{
v=e[i].ed;
if(v==fa)continue;
father[v]=u;
dep[v]=dep[u]+1;
a[v]=e[i].w;
find_son(v,u);
sz[u]+=sz[v];
if(sz[v]>sz[son[u]])son[u]=v;
}
}
void find_time(int u,int fa)
{
int i,v;
ti[u]=idx++;
cot[ti[u]]=a[u];
top[u]=fa;
if(son[u]!=0)find_time(son[u],top[u]);
for(i=head[u];i!=-1;i=e[i].next)
{
v=e[i].ed;
if(v==son[u]||v==father[u])continue;
find_time(v,v);
}
}
//***********************线段树*********************
struct Tree
{
int L,R,Mw,mw;
int flag;//该区间是否取反
}T[N*10];
void up(int id)
{
int li=id<<1,ri=li|1;
T[id].Mw=max(T[li].Mw,T[ri].Mw);
T[id].mw=min(T[li].mw,T[ri].mw);
}
void buildTree(int L,int R,int id)
{
T[id].L=L;T[id].R=R;T[id].flag=0;
if(L==R)
{
T[id].Mw=T[id].mw=cot[L];
return;
}
int mid=(L+R)>>1,li=id<<1,ri=li|1;
buildTree(L,mid,li);
buildTree(mid+1,R,ri);
up(id);
}
void fan(int id)
{
if(T[id].L==T[id].R)return ;
int li=id<<1,ri=li|1;
T[id].flag=0;//传递给两个子区间后该区间上不取反
T[li].flag^=1;T[ri].flag^=1;
T[li].Mw*=-1;T[li].mw*=-1;
T[ri].Mw*=-1;T[ri].mw*=-1;
swap(T[li].Mw,T[li].mw);
swap(T[ri].Mw,T[ri].mw);
}
void Negate(int L,int R,int id)//取反操作
{
if(T[id].L==L&&T[id].R==R)
{
T[id].Mw*=-1;
T[id].mw*=-1;
swap(T[id].Mw,T[id].mw);
T[id].flag^=1;
return ;
}
int mid=(T[id].L+T[id].R)>>1,li=id<<1,ri=li|1;
if(T[id].flag)
{
fan(id);
}
if(R<=mid)Negate(L,R,li);
else if(L>mid)Negate(L,R,ri);
else
{
Negate(L,mid,li);
Negate(mid+1,R,ri);
}
up(id);
}
void insert(int x,int id,int w)//更新x的值
{
if(T[id].L==x&&T[id].R==x)
{
T[id].Mw=T[id].mw=w;
T[id].flag=0;
return ;
}
int mid=(T[id].L+T[id].R)>>1,li=id<<1,ri=li|1;
if(T[id].flag)
fan(id);
if(x<=mid)insert(x,li,w);
else insert(x,ri,w);
up(id);
}
int find(int L,int R,int id)//找最大值
{
if(T[id].L==L&&T[id].R==R)
{
return T[id].Mw;
}
int mid=(T[id].L+T[id].R)>>1,li=id<<1,ri=li|1;
if(T[id].flag)
fan(id);
if(R<=mid)return find(L,R,li);
else if(L>mid)return find(L,R,ri);
else return max(find(L,mid,li),find(mid+1,R,ri));
up(id);
}
int lca(int x,int y)//x到y路径上的最大值
{
int ans=-inf;
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]])swap(x,y);
ans=max(ans,find(ti[top[x]],ti[x],1));
x=father[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
if(x!=y)
ans=max(ans,find(ti[x]+1,ti[y],1));
return ans;
}
void LCA(int x,int y)//x到y路径权值取反
{
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]])swap(x,y);
Negate(ti[top[x]],ti[x],1);
x=father[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
if(x!=y)
Negate(ti[x]+1,ti[y],1);//ti[x]是x与父节点的边
}
int main()
{
int i,n,t,x,y,w;
char str[10];
scanf("%d",&t);
while(t--)
{
memset(head,-1,sizeof(head));
num=0;
scanf("%d",&n);
for(i=1;i<n;i++)
{
scanf("%d%d%d",&x,&y,&w);
addedge(x,y,w);
}
dep[1]=1;
sz[0]=father[1]=0;idx=1;
find_son(1,0);
find_time(1,1);
buildTree(1,n,1);
while(scanf("%s",str),str[0]!='D')
{
scanf("%d%d",&x,&y);
if(str[0]=='C')
{
x=x*2-2;
if(dep[e[x].st]>dep[e[x].ed])//每条边的取值在度数大的点上
swap(e[x].st,e[x].ed);
insert(ti[e[x].ed],1,y);
}
else if(str[0]=='N')
LCA(x,y);
else
printf("%d\n",lca(x,y));
}
}
return 0;
}