Housewife Wind POJ2763 （树边权修改+链长查询 ： dfs序+树状数组+lca)

大致题意

给定一颗有边权的树，支持两种操作，修改某条边的边权，查询从s到某个点距离，初始s的位置给定，后面跟新。

思路

以前我很智障的以为带修改的树上链长只能用树剖做，dfs序只能做子树和…然后最近补网络赛突然发现，修改边权可以把边的权值放到深度较大的点上，因为从根到其他节点的长度，只有这条边深度较深的点以下的子树会受到修改边权的影响。本身从根到叶子的距离就是一个距离的前缀和，把子树映射到dfs序，相当于差分数组的区间修改，查询前缀和，可以用树状数组维护。我们都知道链长dis(x,y) = dis(rt,x) + dis(rt,y) - 2*dis(rt,lca(x,y))

代码

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<ctype.h>
using namespace std;
#define maxn 100005
#define maxm 1000006
#define ll long long int
#define INF 0x3f3f3f3f
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
#define mem(a) memset(a,0,sizeof(a))
#define sqr(x) (x*x)
#define inf (ll)2e18+1
#define PI acos(-1)
int read(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
     while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
     return f*x;
}
int n,m,s;
struct edge{int u,v,w,nxt;}ed[maxn<<1];
int head[maxn],tot;
void add(int x,int y,int z){
    ed[++tot]={x,y,z,head[x]};head[x]=tot;
}
int in[maxn],out[maxn],dep[maxn],dis[maxn],fa[maxn][21],cnt,c[maxn];
void dfs(int x,int y){
    in[x]=++cnt;
    dep[x]=dep[y]+1;
    fa[x][0]=y;
    inc(i,1,20)fa[x][i]=fa[fa[x][i-1]][i-1];
    for(int i=head[x];i;i=ed[i].nxt){
        int v=ed[i].v;
        if(v==y)continue;
        dis[v]=dis[x]+ed[i].w;
        dfs(v,x);
    }
    out[x]=cnt;
}
int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    dec(i,20,0)if(dep[fa[x][i]]>=dep[y])x=fa[x][i];
    if(x==y)return x;
    dec(i,20,0)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
void add(int x,int y){
    while(x<maxn){
        c[x]+=y;
        x+=x&(-x);
    }
}
int ask(int x){
    int res=0;
    while(x){
        res+=c[x];
        x-=x&(-x);
    }
    return res;
}
int main()
{
    n=read();m=read();s=read();
    int x,y,z;
    inc(i,1,n-1){
        x=read();y=read();z=read();
        add(x,y,z);add(y,x,z);
    }
    dfs(s,0);
    while(m--){
        z=read();
        if(z==0){
            x=read();
            int p=lca(x,s);
            int ans=ask(in[x])+ask(in[s])-2*ask(in[p])+dis[x]+dis[s]-2*dis[p];
            printf("%d\n",ans);
            s=x;
        }
        else {
            x=read();y=read();
            int u=ed[x*2-1].u;
            int v=ed[x*2-1].v;
            int w=ed[x*2-1].w;
            if(dep[u]>dep[v]){
                add(in[u],y-w);add(out[u]+1,w-y);
            }
            else {
                add(in[v],y-w);add(out[v]+1,w-y);
            }
            ed[x*2-1].w=y;
        }
    }
    return 0;
}