POJ2763 Housewife Wind（DFS序）

题目：单边修改，树链查询。

这题是边权，不是点权，不过也可以看作是点权。

然后其实就和BZOJ2819一样。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 111111
struct Edge{
    int u,v,w,nxt;
}edge[MAXN<<1];
int n,head[MAXN],NE;
void addEdge(int u,int v,int w){
    edge[NE].u=u; edge[NE].v=v; edge[NE].w=w; edge[NE].nxt=head[u]; head[u]=NE++;
}
int odr,l[MAXN],r[MAXN],stack[MAXN],sum[MAXN],fa[20][MAXN],dep[MAXN];
void dfs(){
    int top=0;
    stack[++top]=1;
    while(top){
        int u=stack[top];
        if(l[u]){
            r[u]=odr; --top;
            continue;
        }
        l[u]=++odr;
        for(int i=head[u]; i!=-1; i=edge[i].nxt){
            int v=edge[i].v;
            if(fa[0][u]==v) continue;
            fa[0][v]=u; dep[v]=dep[u]+1; sum[v]=sum[u]+edge[i].w;
            stack[++top]=v;
        }
    }
}

int N,tree[MAXN<<2],x,y,z;
void update(int i,int j,int k){
    if(x<=i && j<=y){
        tree[k]+=z;
        return;
    }
    if(tree[k]){
        tree[k<<1]+=tree[k]; tree[k<<1|1]+=tree[k];
        tree[k]=0;
    }
    int mid=i+j>>1;
    if(x<=mid) update(i,mid,k<<1);
    if(y>mid) update(mid+1,j,k<<1|1);
}
int query(int i,int j,int k){
    if(i==j) return tree[k];
    if(tree[k]){
        tree[k<<1]+=tree[k]; tree[k<<1|1]+=tree[k];
        tree[k]=0;
    }
    int mid=i+j>>1;
    if(x<=mid) return query(i,mid,k<<1);
    return query(mid+1,j,k<<1|1);
}
int lca(int u,int v){
    if(dep[u]>dep[v]) swap(u,v);
    for(int k=0; k<20; ++k){
        if((dep[v]-dep[u])>>k&1){
            v=fa[k][v];
        }
    }
    if(v==u) return u;
    for(int k=19; k>=0; --k){
        if(fa[k][u]!=fa[k][v]){
            u=fa[k][u];
            v=fa[k][v];
        }
    }
    return fa[0][u];
}
void init(){
    dfs();
    for(int i=1; i<20; ++i){
        for(int j=1; j<=n; ++j){
            int t=fa[i-1][j];
            fa[i][j]=fa[i-1][t];
        }
    }
    for(N=1; N<odr; N<<=1);
    for(int i=1; i<=n; ++i){
        x=l[i]; y=l[i]; z=sum[i];
        update(1,N,1);
    }
}

int main(){
    int q,s,op,a,b,c;
    memset(head,-1,sizeof(head));
    scanf("%d%d%d",&n,&q,&s);
    for(int i=1; i<n; ++i){
        scanf("%d%d%d",&a,&b,&c);
        addEdge(a,b,c);
        addEdge(b,a,c);
    }
    init();
    while(q--){
        scanf("%d",&c);
        if(c){
            scanf("%d%d",&a,&b);
            Edge &e=edge[a-1<<1];
            if(fa[0][e.u]==e.v){
                x=l[e.u]; y=r[e.u]; z=b-e.w;
            }else{
                x=l[e.v]; y=r[e.v]; z=b-e.w;
            }
            e.w=b;
            update(1,N,1);
        }else{
            scanf("%d",&a);
            int res;
            x=l[a]; res=query(1,N,1);
            x=l[s]; res+=query(1,N,1);
            x=l[lca(a,s)]; res-=(query(1,N,1)<<1);
            s=a;
            printf("%d
",res);
        }
    }
    return 0;
}