bzoj3991 [Sdoi2015]寻宝游戏 set动态维护虚树+树链求并

题目大意：支持多次操作，增加或删除一个关键点

动态维护虚树边权和*2

分析：可以用树链求并的方法，最后减去虚树的根到1距离

注意到树链求并是所有点到根距离-所有dfn序相邻两点的LCA到根距离

找dfn左右两个点稍微求一下，那就平衡树维护咯

树总根是dfn最小的点和最大的点的LCA

=================================================

md智障想偷懒不写平衡树，写了个set，然后gg不会写搞了两个小时

=================================================

set定义+迭代器

set<int>S;
set<int>::iterator it;

set存储方式：

1.左闭右开

begin()是指向第一个点,end()指向最后一个点后的一个结束指针

2.特别的，set空时begin()==end()

3.不允许有重复元素

set小函数（方便使用）

int getnxt(int x){
    it=S.upper_bound(x);
    return it!=S.end() ? pid[*it] : -1;
}
int getpre(int x){
    it=S.lower_bound(x);
    return it!=S.begin() ? pid[*(--it)] : -1;
}
int getfirst(){
    it=S.begin();
    return it!=S.end() ? pid[*it] : -1;
}
int getlast(){
    it=S.end();
    return it!=S.begin() ? pid[*(--it)] : -1;
} 

本题代码

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cctype>
#include <set>
using namespace std;
typedef long long LL;
const int M=100007;
inline int rd(){
    int x=0;bool f=1;char c=getchar();
    for(;!isdigit(c);c=getchar())if(c=='-')f=0;
    for(;isdigit(c);c=getchar())x=x*10+c-48;
    return f?x:-x;
}

int vis[M];
int n,m;
bool use[M];
int g[M],te;
struct edge{int y,next;LL d;}e[M<<1];
void addedge(int x,int y,LL d){
    e[++te].y=y;e[te].d=d;e[te].next=g[x];g[x]=te;
}
int dfn[M],pid[M],tdfn;
int a[M<<1][20],ln[M<<1],pos[M],T;
LL dis[M];

void dfs(int x,int fa){
    pid[dfn[x]=++tdfn]=x;
    a[pos[x]=++T][0]=x;
    int p,y;
    for(p=g[x];p;p=e[p].next)
    if((y=e[p].y)!=fa){
        dis[y]=dis[x]+e[p].d;
        dfs(y,x);
        a[++T][0]=x;
    }
}

set<int>S;
set<int>::iterator it;

int mn(int x,int y){return dfn[x]<dfn[y]?x:y;}

void init(){
    int i,j,l;
    for(i=2;i<=T;i++) ln[i]=ln[i>>1]+1;
    for(i=T;i>0;i--){
        l=ln[T-i+1];
        for(j=1;j<=l;j++) a[i][j]=mn(a[i][j-1],a[i+(1<<j-1)][j-1]);
    }
}

int LCA(int x,int y){
    x=pos[x],y=pos[y];
    int l=ln[y-x+1];
    return mn(a[x][l],a[y-(1<<l)+1][l]);
}

int getnxt(int x){
    it=S.upper_bound(x);
    return it!=S.end() ? pid[*it] : -1;
}

int getpre(int x){
    it=S.lower_bound(x);
    return it!=S.begin() ? pid[*(--it)] : -1;
}

int getfirst(){
    it=S.begin();
    return it!=S.end() ? pid[*it] : -1;
}

int getlast(){
    it=S.end();
    return it!=S.begin() ? pid[*(--it)] : -1;
} 

int main(){
    freopen("a.txt","r",stdin);
    int i,x,y,z,l,r;
    LL ans=0;
    n=rd(),m=rd();
    for(i=1;i<n;i++){
        x=rd(),y=rd(),z=rd();
        addedge(x,y,z);
        addedge(y,x,z);
    }
    dfs(1,0);
    init();
    for(i=1;i<=m;i++){
        x=rd();
        if(use[x]){
            S.erase(dfn[x]);
            l=getpre(dfn[x]);
            r=getnxt(dfn[x]);

            if(l!=-1&&r!=-1) ans-=dis[LCA(l,r)];
            if(l!=-1) ans+=dis[LCA(l,x)];
            if(r!=-1) ans+=dis[LCA(x,r)];
            ans-=dis[x];
        }
        else{
            l=getpre(dfn[x]);
            r=getnxt(dfn[x]);
            S.insert(dfn[x]);

            if(l!=-1&&r!=-1) ans+=dis[LCA(l,r)];
            if(l!=-1) ans-=dis[LCA(l,x)];
            if(r!=-1) ans-=dis[LCA(x,r)];
            ans+=dis[x];
        }
        use[x]^=1;

        it=S.begin();
        if(it==S.end()) puts("");
        else{
            l=getfirst();
            r=getlast();
            printf("%lld\n",(ans-dis[LCA(l,r)])*2);
        }
    }
    return 0;
}