bzoj3991 [SDOI2015]寻宝游戏（虚树+set）

动态删减关键点，我们发现答案就是虚树的边长和*2.
而虚树的边长和二倍正好就是把所有关键点按dfs序排序后相邻两点间的距离和。（包括最后一个和第一个的距离和）

因此我们用set维护一下即可。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(T==S){T=(S=buf)+fread(buf,1,1<<16,stdin);if(S==T) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,h[N],num=0,fa[N][20],dfn[N],dfnum=0,dep[N],Log[N];
ll dis[N],ans=0;bool inq[N];
struct cmp{
    bool operator()(int x,int y){return dfn[x]<dfn[y];}
};
set<int,cmp>st;set<int,cmp>::iterator it;
struct edge{
    int to,next,val;
}data[N<<1];
void dfs(int x){
    dfn[x]=++dfnum;
    for(int i=1;i<=Log[n];++i){
        if(!fa[x][i-1]) break;
        fa[x][i]=fa[fa[x][i-1]][i-1];
    }for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x][0]) continue;
        fa[y][0]=x;dep[y]=dep[x]+1;dis[y]=dis[x]+data[i].val;dfs(y);
    }
}
inline int lca(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    int d=dep[x]-dep[y];
    for(int i=0;i<=Log[d];++i)
        if(d>>i&1) x=fa[x][i];
    if(x==y) return x;
    for(int i=Log[n];i>=0;--i)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
inline ll caldis(int x,int y){
    return dis[x]+dis[y]-2*dis[lca(x,y)];
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();Log[0]=-1;
    for(int i=1;i<=n;++i) Log[i]=Log[i>>1]+1;
    for(int i=1;i<n;++i){
        int x=read(),y=read(),val=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].val=val;
        data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].val=val;
    }dfs(1);
    while(m--){
        int x=read();
        if(inq[x]){
            st.erase(x);if(st.empty()) ans=0;
            else{
                int pre,succ;it=st.lower_bound(x);
                if(it==st.end()) succ=*(st.begin());
                else succ=*it;
                if(it==st.begin()) pre=*(st.rbegin());
                else pre=*(--it);ans-=caldis(x,pre)+caldis(x,succ);
                ans+=caldis(pre,succ);
            }
        }else{
            if(!st.empty()){
                int pre,succ;it=st.lower_bound(x);
                if(it==st.end()) succ=*(st.begin());
                else succ=*it;
                if(it==st.begin()) pre=*(st.rbegin());
                else pre=*(--it);ans+=caldis(x,pre)+caldis(x,succ);
                ans-=caldis(pre,succ);
            }st.insert(x);
        }printf("%lld\n",ans);inq[x]^=1;
    }return 0;
}