poj 3237 树链剖分

这几天，看到别的队都有会树链剖分的，于是就来学了一下。。。

就是把边的权值变成了深度较深的那个点的权值

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;

const int maxn = 100010;
int sum[maxn*4],add[maxn*4];
struct si{
    int f,t,c,next;
}side[maxn*2];
int head[maxn];
int tid[maxn],ran[maxn],num[maxn],son[maxn];
int top[maxn],fa[maxn],siz[maxn],dep[maxn];
int tim,cnt;
int n,q,s;

void dfs1(int u,int father,int d)
{
    dep[u]=d;
    fa[u]=father;
    siz[u]=1;
    for(int i=head[u]; i!=-1; i=side[i].next)
    {
        int v=side[i].t;
        if(v!=father)
        {
            dfs1(v,u,d+1);
            siz[u]+=siz[v];
            if(son[u]==-1||siz[v]>siz[son[u]])
                son[u]=v;
        }
    }
}

void dfs2(int u,int tp)
{
    //cout<<888888<<endl;
    top[u]=tp;
    tid[u]=++tim;
    ran[tid[u]]=u;
    if(son[u]==-1) return;
    dfs2(son[u],tp);
    for(int i=head[u]; i!=-1; i=side[i].next)
    {   //cout<<i<<endl;
        int v=side[i].t;
        //cout<<v<<endl;
        if(v!=son[u]&&v!=fa[u])
            dfs2(v,v);
    }
}

void init(){
    tim=cnt=0;
    memset(head,-1,sizeof(head));
    memset(son,-1,sizeof(son));
}

void addside(int f,int t,int c){
    side[cnt]=(si){f,t,c,head[f]};
    head[f]=cnt++;
    side[cnt]=(si){t,f,c,head[t]};
    head[t]=cnt++;
}


/*
void change(int a,int b,int c){
    while(top[a]!=top[b]){
        if(dep[top[a]]<dep[top[b]]) swap(a,b);
        update(tid[top[a]],tid[a],c,1,n,1);
        a=fa[top[a]];
    }
    if(dep[a]<dep[b]) swap(a,b);
    update(tid[b],tid[a],c,1,n,1);
}
*/

void build(int l,int r,int rt){
    if(l==r){
        sum[rt]=num[l];
        return ;
    }
    int mid=(l+r)/2;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void update(int L,int R,int c,int l,int r,int rt){
    if(l>=L&&r<=R){
        sum[rt]=c;
        return ;
    }
    int mid=(l+r)/2;
    if(mid>=L) update(L,R,c,l,mid,rt<<1);
    if(mid+1<=R) update(L,R,c,mid+1,r,rt<<1|1);
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void change(int a,int b){
    int f=side[a*2-1].f,t=side[a*2-1].t;
    if(dep[f]<dep[t]) swap(f,t);
    update(tid[f],tid[f],b,2,n,1);
}

int query(int L,int R,int l,int r,int rt){
    int ans=0;
    if(l>=L&&r<=R){
        return sum[rt];
    }
    int mid=(l+r)/2;
    if(mid>=L) ans+=query(L,R,l,mid,rt<<1);
    if(mid+1<=R) ans+=query(L,R,mid+1,r,rt<<1|1);
    return ans;
}

void Query(int a,int b){
    int ans=0;
    while(top[a]!=top[b]){
        if(dep[top[a]]<dep[top[b]]) swap(a,b);
        ans+=query(tid[top[a]],tid[a],2,n,1);
        a=fa[top[a]];
    }
    if(dep[a]>dep[b]) swap(a,b);
    if(a!=b) ans+=query(tid[a]+1,tid[b],2,n,1);
    printf("%d\n",ans);
}

int main()
{
    while(cin>>n>>q>>s){
        init();
        int a,b,c;
        for(int i=0;i<n-1;i++){
            scanf("%d %d %d",&a,&b,&c);
            addside(a,b,c);
        }
        dfs1(1,1,1);
        dfs2(1,1);
        for(int i=0;i<cnt;i+=2){
            int f=side[i].f, t=side[i].t, c=side[i].c;
            if(dep[f]>dep[t]) num[tid[f]]=c;
            else num[tid[t]]=c;
        }
        build(2,n,1);
        while(q--){
            scanf("%d",&a);
           // cout<<a<<endl;
            if(a==0){
                scanf("%d",&b);
                //cout<<a<<' '<<b<<endl;
                Query(b,s);
                s=b;
            }
            if(a==1){
                scanf("%d %d",&b,&c);
                //cout<<a<<' '<<b<<' '<<c<<endl;
                change(b,c);
            }
        }
    }
    return 0;
}