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

bzoj3991 [SDOI2015]寻宝游戏

原题地址：http://www.lydsy.com/JudgeOnline/problem.php?id=3991

题意：
一棵n个节点的树，m个操作，每次取消或添加一个关键点，每次操作输出求包含当前这些关键点虚树的路径和的两倍。

数据范围
1<=N<=100000

1<=M<=100000

对于全部的数据，1<=z<=10^9

题解：
（今天的t3）

答案为按DFS序排序后相邻两个关键点的间距离和加上首尾两个点的距离。
于是乎把所有的关键点塞进set里，维护一个sum。
set中在其前一个的为pre，在其后一个的为nxt
删除一个点就加上pre与nxt之间的距离，减去其与pre，其与nxt的距离。
加入一个点就减去pre与nxt之间的距离，加上其与pre，其与nxt的距离。
求距离就简单地求下lca。
注意答案还要加上首尾两个点的距离，另外为了方便处理没有pre或没有nxt的情况，可以先往set中加入0和n+1便于特判。

（get了访问set中前/后一个元素的方法：int pre=* –S.find(dfn[x]); int nxt=* ++S.find(dfn[x]);）
（现在知道虚树为何，膜你抄全歌词达成☆）

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<set>
#define LL long long
using namespace std;
const int N=200005;
const int P=18;
set<int> S;
int n,m,head[N],to[2*N],nxt[2*N],num=0;
int dep[N],dfn[N],seq[N],inc=0,anc[N][P+5];
LL w[2*N],dis[N];
bool vis[N];
LL sum=0;
void build(int u,int v,LL ww)
{
    num++;
    to[num]=v;
    nxt[num]=head[u];
    w[num]=ww;
    head[u]=num;
}
void dfs(int u,int f)
{
    inc++; 
    anc[u][0]=f;
    for(int i=1;i<P;i++)
    anc[u][i]=anc[anc[u][i-1]][i-1];
    seq[inc]=u; dfn[u]=inc;
    dep[u]=dep[f]+1;
    for(int i=head[u];i;i=nxt[i])
    {
        int v=to[i];
        if(v==f) continue;
        dis[v]=dis[u]+w[i];
        dfs(v,u);
    }
}
int getlca(int u,int v)
{
    if(dep[u]<dep[v]) swap(u,v);

    int d=dep[u]-dep[v];
    for(int i=0;d;d>>=1,i++)
    if(d&1) u=anc[u][i];
    if(u==v) return u;
    for(int i=P-1;i>=0;i--)
    if(anc[u][i]!=anc[v][i])
    {u=anc[u][i]; v=anc[v][i];}
    return anc[u][0];

}
LL cul(int u,int v)
{   
    int lca=getlca(u,v);
    return dis[u]+dis[v]-2*dis[lca];
}
int main()
{
    memset(head,0,sizeof(head));
    memset(vis,0,sizeof(vis));
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)
    {
        int u,v; LL ww;
        scanf("%d%d%lld",&u,&v,&ww);
        build(u,v,ww); build(v,u,ww);
    }
    dfs(1,1);
    S.insert(0); S.insert(n+1);
    while(m--)
    {
        int x;
        scanf("%d",&x);
        if(!vis[x])
        {
            S.insert(dfn[x]);
            int pre=*--S.find(dfn[x]);
            int nxt=*++S.find(dfn[x]);
            if(pre>=1&&nxt<=n)
            sum-=cul(seq[pre],seq[nxt]);
            if(pre>=1) sum+=cul(seq[pre],x);
            if(nxt<=n) sum+=cul(x,seq[nxt]);
            vis[x]=1;
        }
        else
        {
            int pre=*--S.find(dfn[x]);
            int nxt=*++S.find(dfn[x]);
            if(pre>=1&&nxt<=n)
            sum+=cul(seq[pre],seq[nxt]);
            if(pre>=1) sum-=cul(seq[pre],x);
            if(nxt<=n) sum-=cul(x,seq[nxt]);
            vis[x]=0;
            S.erase(dfn[x]);
        }
        LL add;
        int first=*++S.find(0); int last=*--S.find(n+1);
        if(first>n||last<1||first==last) add=0;
        else add=cul(seq[first],seq[last]);
        printf("%lld\n",sum+add);
    }
    return 0;
}