SPOJ Query on a tree 树链剖分 边修改

链接

提交链接

题解

对边的修改算到点上
只需要修改下面的地方

代码

#include<bits/stdc++.h>
#define N 10010
#define INF 0x3f3f3f3f
#define eps 1e-10
#define pi 31592653589793
#define P 1000000007 
#define LL long long
#define pb push_back
#define fi first
#define se second
#define cl clear
#define si size
#define lb lower_bound
#define ub upper_bound
#define mem(x) memset(x,0,sizeof x)
#define sc(x) scanf("%d",&x)
#define scc(x,y) scanf("%d%d",&x,&y)
#define sccc(x,y,z) scanf("%d%d%d",&x,&y,&z)
using namespace std;
vector<int> a[N];
int n,m,cnt,rt,tg,son[N],top[N],id[N],fa[N],d[N],sz[N],rk[N],w[N],mod;
int f[N<<2];
void dfs1(int x,int ffa){
    sz[x]=1;
    for(auto i:a[x]) if (i!=ffa){
        fa[i]=x; d[i]=d[x]+1; dfs1(i,x); sz[x]+=sz[i];
        if (sz[i]>sz[son[x]]) son[x]=i;
    }
}
void dfs2(int x,int t){
    top[x]=t; id[x]=++cnt; rk[cnt]=x;
    if (!son[x]) return;
    dfs2(son[x],t);
    for (auto i:a[x]) if (i!=son[x] && i!=fa[x]) dfs2(i,i);
}


void updata(int x,int l,int r,int p){
    if (l==r){
        f[x]=tg;
    }else{
        int t=l+r>>1;
        if (p<=t) updata(x<<1,l,t,p);else
            updata(x<<1|1,t+1,r,p);
        f[x]=max(f[x<<1],f[x<<1|1]);
    }
}

int query(int x,int l,int r,int fl,int fr){
    if (l==fl && r==fr) return f[x];
    int t=l+r>>1;
    if (fr<=t)return query(x<<1,l,t,fl,fr);else
    if (fl>t)return query(x<<1|1,t+1,r,fl,fr);else
    return max(query(x<<1,l,t,fl,t),query(x<<1|1,t+1,r,t+1,fr));
}

int sum(int x,int y){
    int ans=0;
    while(top[x]!=top[y]){
        if (d[top[x]]<d[top[y]]) swap(x,y);
        ans=max(ans,query(1,1,n,id[top[x]],id[x]));
        x=fa[top[x]];
    }
    if(x==y) return ans;
    if (d[x]>d[y]) swap(x,y);
    ans=max(ans,query(1,1,n,id[x]+1,id[y]));
    return ans;
}

int e[N][3];
int main(){
    int T;
    sc(T);
    while(T--){
        sc(n);
        for (int i=1;i<=n;i++) a[i].clear(),son[i]=0;
        for (int i=1;i<=n*4;i++) f[i]=0;
        cnt=0;
        for (int i=1,x,y,z;i<n;i++){
            sccc(x,y,z);
            a[x].pb(y),a[y].pb(x);
            e[i][0]=x;e[i][1]=y;e[i][2]=z;
        }
        dfs1(1,-1);
        dfs2(1,1);
        for(int i=1;i<n;i++){
            if (d[e[i][0]]<d[e[i][1]]) swap(e[i][0],e[i][1]);
            tg=e[i][2];
            updata(1,1,n,id[e[i][0]]);
        }

        char ch[10];
        scanf("%s",ch);
        while(ch[0]!='D'){
            int x,y;
            scc(x,y);
            if (ch[0]=='Q'){
                printf("%d\n",sum(x,y));
            }else{
                tg=y;
                updata(1,1,n,id[e[x][0]]);
            }
            scanf("%s",ch);
        }
    }
}