BZOJ 3991 set维护dfs序

思路：

set按照dfn排序

两点之间的距离可以O(logn)算出来

加一个点-> now

ans+=dis(pre,now)+dis(now,next)-dis(pre-next);

删一个点同理

最后加上dis(begin,end)即可

//By SiriusRen
#include <set>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=200050;
typedef long long ll;
int first[N],next[N],v[N],w[N],tot,cnt,rev[N];
int n,m,xx,yy,zz,dfn[N],deep[N],fa[N/2][20];
ll ans,wei[N],temp;
struct cmp{bool operator()(int x,int y){return dfn[x]<dfn[y];}};
set<int,cmp>s;set<int,cmp>::iterator it,itpre,itnext;
void add(int x,int y,int z){w[tot]=z,v[tot]=y,next[tot]=first[x],first[x]=tot++;}
void dfs(int x){
    dfn[x]=++cnt;
    for(int i=first[x];~i;i=next[i])
        if(v[i]!=fa[x][0])
            deep[v[i]]=deep[x]+1,wei[v[i]]=wei[x]+w[i],fa[v[i]][0]=x,dfs(v[i]);
}
int lca(int x,int y){
    if(deep[x]<deep[y])swap(x,y);
    for(int i=19;i>=0;i--)if(deep[x]-(1<<i)>=deep[y])x=fa[x][i];
    if(x==y)return x;
    for(int i=19;i>=0;i--)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
ll dis(int x,int y){return wei[x]+wei[y]-2*wei[lca(x,y)];}
int main(){
    memset(first,-1,sizeof(first));
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&xx,&yy,&zz);
        add(xx,yy,zz),add(yy,xx,zz);
    }
    dfs(1),deep[1]=1;
    for(int j=1;j<=19;j++)
        for(int i=1;i<=n;i++)
            fa[i][j]=fa[fa[i][j-1]][j-1];
    while(m--){
        scanf("%d",&xx);
        rev[xx]^=1;
        if(rev[xx]){
            s.insert(xx),it=s.lower_bound(xx);
            itnext=++it,--it;
            if(it!=s.begin())itpre=--it,++it,ans+=dis(*itpre,*it);
            if(itnext!=s.end())ans+=dis(*itnext,*it);
            if(it!=s.begin()&&itnext!=s.end())ans-=dis(*itpre,*itnext);
        }
        else{
            it=s.lower_bound(xx);
            itnext=++it,--it;
            if(it!=s.begin())itpre=--it,++it,ans-=dis(*itpre,*it);
            if(itnext!=s.end())ans-=dis(*itnext,*it);
            if(it!=s.begin()&&itnext!=s.end())ans+=dis(*itpre,*itnext);
            s.erase(it);
        }
        it=s.end();
        if(s.size()>1)temp=dis(*s.begin(),*(--it));
        else temp=0;
        printf("%lld\n",ans+temp);
    }
}