spoj 375 树链剖分模板

/*只是一道树链刨分的入门题,作为模板用。*/#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 0x3fffffffint head[N];int son[N];//记录与当前点相连的数目最多的子节点的下标int fa[N];//记录上一个父节点int siz[N];//记录当前节点的子节点的数目int top[N];//当前链的最顶层int f[N];//重新标记int fp[N];//记录重新标记前的点int deep[N];//深度int w[N];//  记录当前点与其父节点的关系int nu,yong;int Max;struct nodee{    int u,v,w,next;} bian[N*4],ff[N];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])//排除重链            getnu(v,v);//轻链    }    return;}//******以上求重链和轻链以及各部分的初始化，下面是线段树部分和查询询问**********//struct node{    int l,r,maxx;} tree[N*4];int Ma(int v,int vv){    return v>vv?v:vv;}void pushup(int t) //回溯时更新最大值和最小值{    tree[t].maxx=Ma(tree[t*2].maxx,tree[t*2+1].maxx);}void build(int t,int l,int r)//建树{    tree[t].l=l;    tree[t].r=r;    if(tree[t].l==tree[t].r)    {        tree[t].maxx=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;    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=y;        return ;    }    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);}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                update(1,f[ff[x].u],y);//更换区间中某个值,ff[x].u是深度较大的数，所以不会出现越界情况，即f[ff[x].u]不为0        }    }    return 0;}