poj 3237 树链剖分模板（用到线段树lazy操作）

/*
本体在spoj375的基础上加了一些操作，用到线段树的lazy操作模板类型
*/
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<map>
#include<string.h>
#include<stdlib.h>
#include<math.h>
using namespace std;
#define N  11000
#define inf 0x3fffffff
int head
;
int son
;//记录与当前点相连的数目最多的子节点的下标
int fa
;//记录上一个父节点
int siz
;//记录当前节点的子节点的数目
int top
;//当前链的最顶层
int f
;//重新标记
int fp
;//记录重新标记前的点
int deep
;//深度
int w
;//  记录当前点与其父节点的关系
int nu,yong;
int Max;
struct nodee
{
int u,v,w,next;
} bian[N*4],ff
;
void init()
{
yong=nu=0;
memset(head,-1,sizeof(head));
memset(son,-1,sizeof(son));
}
void addedge(int u,int v,int w)
{
bian[yong].u=u;
bian[yong].v=v;
bian[yong].next=head[u];
head[u]=yong++;
}
void dfs(int u,int father,int d)
{
deep[u]=d;//记录深度
siz[u]=1;//初始化
fa[u]=father;//记录父节点
int i;
for(i=head[u]; i!=-1; i=bian[i].next)
{
int v=bian[i].v;
if(v!=father)
{
dfs(v,u,d+1);//
siz[u]+=siz[v];//回溯时累加数目
if(son[u]==-1||siz[son[u]]<siz[v])//son[u]记录当前点相连的点的子节点数目最多的点，即重链
son[u]=v;//u-v
}
}
}
void getnu(int u,int cnt)
{
top[u]=cnt;//记录当前链的顶端
f[u]=nu++;//重新标记
fp[f[u]]=u;//记录标记前的编号
if(son[u]==-1)return ;//如果他没有儿子节点
getnu(son[u],cnt);//重链
int i;
for(i=head[u]; i!=-1; i=bian[i].next)
{
int v=bian[i].v;
if(v!=fa[u]&&v!=son[u])//排除与u相连的重链
getnu(v,v);//轻链
}
return;
}
//******以上求重链和轻链以及各部分的初始化，下面是线段树部分和查询询问**********//
struct node
{
int l,r,maxx,yanchi,minn;
} tree[N*4];
int Ma(int v,int vv)
{
return v>vv?v:vv;
}
int Min(int v,int vv) {
return v>vv?vv:v;
}
void pushdown(int t) {
while(tree[t].yanchi) {
tree[t*2].maxx=-tree[t*2].maxx;
tree[t*2].minn=-tree[t*2].minn;
swap(tree[t*2].maxx,tree[t*2].minn);
tree[t*2].yanchi^=1;
tree[t*2+1].maxx=-tree[t*2+1].maxx;
tree[t*2+1].minn=-tree[t*2+1].minn;
swap(tree[t*2+1].maxx,tree[t*2+1].minn);
tree[t*2+1].yanchi^=1;
tree[t].yanchi=0;
}
return ;
}
void pushup(int t) //回溯时更新最大值和最小值
{
tree[t].maxx=Ma(tree[t*2].maxx,tree[t*2+1].maxx);
tree[t].minn=Min(tree[t*2].minn,tree[t*2+1].minn);
}
void build(int t,int l,int r)//建树
{
tree[t].l=l;
tree[t].r=r;
tree[t].yanchi=0;
if(tree[t].l==tree[t].r)
{
tree[t].maxx=tree[t].minn=w[tree[t].l];//记录边权值
return ;
}
int mid=(l+r)>>1;
build(t*2,l,mid);
build(t*2+1,mid+1,r);
pushup(t);
}
void qury(int t,int l,int r)//询问区间的最大值
{
if(tree[t].l==l&&tree[t].r==r)//如果查到
{
Max=Ma(Max,tree[t].maxx);//
return ;
}
int mid=(tree[t].l+tree[t].r)>>1;
pushdown(t);
if(r<=mid)
qury(t*2,l,r);
else if(l>mid)
qury(t*2+1,l,r);
else
{
qury(t*2,l,mid);
qury(t*2+1,mid+1,r);
}
pushup(t);
}
int findmax(int u,int  v)//查找最大值
{
int f1=top[u];//得到顶端编号值
int f2=top[v];
int ans=-inf;//初始化最小值
while(f1!=f2)//结束条件,再通一个重链上
{
if(deep[f1]<deep[f2])//从最深层开始网上
{
swap(f1,f2);//交换
swap(u,v);
}
Max=-inf;
qury(1,f[f1],f[u]);//询问重新编号后的f[u]和其顶端节点之间的最大值，从而使其从f[u]跳到顶端
ans=Ma(ans,Max);//ans储存最大值
u=fa[f1];//从f1向上跳一步,不管当前链是轻链还是重链
f1=top[u];//得到跳一步后的顶端节点，继续比较
}//
if(v==u)return ans;//如果在同一点就直接返回
if(deep[u]>deep[v])swap(u,v);//得到u，v之间的最小值
Max=-inf;
qury(1,f[son[u]],f[v]);//求出u的子节点的f[v]---f[u的最大数目子节点];,因为此时他们在一个重链上
ans=Ma(ans,Max);//求出最大值
return ans;
}
void update(int t,int x,int y)//更新
{
if(tree[t].l==x&&tree[t].r==x)//
{
tree[t].maxx=tree[t].minn=y;
return ;
}
pushdown(t);
int mid=(tree[t].l+tree[t].r)/2;
if(x<=mid)
update(t*2,x,y);
else
update(t*2+1,x,y);
pushup(t);
}
void chant(int t,int l,int r) {
if(tree[t].l==l&&tree[t].r==r) {
tree[t].maxx=-tree[t].maxx;
tree[t].minn=-tree[t].minn;
swap(tree[t].maxx,tree[t].minn);
tree[t].yanchi^=1;
return ;
}
pushdown(t);
int mid=(tree[t].l+tree[t].r)/2;
if(r<=mid)chant(t*2,l,r);
else if(l>mid)chant(t*2+1,l,r);
else {
chant(t*2,l,mid);
chant(t*2+1,mid+1,r);
}
pushup(t);
}
void change(int u,int v) {
int f1=top[u];
int f2=top[v];
while(f1!=f2) {
if(deep[f1]<deep[f2]) {
swap(f1,f2);
swap(u,v);
}
chant(1,f[f1],f[u]);
u=fa[f1];
f1=top[u];
}
if(u==v)return ;
if(deep[u]>deep[v])swap(u,v);
chant(1,f[son[u]],f[v]);
return ;
}
int main()
{
int T,i,n;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
init();
for(i=1; i<n; i++)
{
scanf("%d%d%d",&ff[i].u,&ff[i].v,&ff[i].w);
addedge(ff[i].u,ff[i].v,ff[i].w);
addedge(ff[i].v,ff[i].u,ff[i].w);
}
dfs(1,1,0);//深搜的到 每个节点的深度，父节点和其子节点的数目（包括本身），还有最大数目的子节点的编号
getnu(1,1);//得到重链或者轻链的顶端和重新编号并记录重新编号前的值。如果是轻链的话
for(i=1; i<n; i++)
{
if(deep[ff[i].u]<deep[ff[i].v])
swap(ff[i].u,ff[i].v);//得到深度最大的节点
w[f[ff[i].u]]=ff[i].w;//记录重新编号后的当前点与上一个点的权值
}
build(1,1,nu-1);//建树
char s[222];
int x,y;
while(scanf("%s",s),strcmp(s,"DONE"))
{
scanf("%d%d",&x,&y);
if(s[0]=='Q')
printf("%d\n",findmax(x,y));//找区间最大值
else
if(s[0]=='C')
update(1,f[ff[x].u],y);//更换区间中某个值,ff[x].u是深度较大的数，所以不会出现越界情况，即f[ff[x].u]不为0
else
change(x,y);
}
}
return 0;
}