bzoj 3991: [SDOI2015]寻宝游戏 dfs序+虚树+set

题意

给出一棵树，每次操作会把一个点放入宝藏或把宝藏拿走，每次操作完后问从任意一个点开始遍历所有宝藏点最后回到起点的最短路径。
n,m<=100000

分析

一开始想到树剖上面去了。。。

其实那个距离就是虚树中任意两点间的距离和*2
同时等于把虚树所有点按照dfs序排好之后相邻两点的距离和加上起点到终点的距离和。
用一个set暴力维护即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<set>
#define N 100005
#define LL long long
using namespace std;

int cnt,n,m,last[N],fa[N][20],dep[N],dfn[N],tim,trea[N];
LL len[N];
struct edge{int to,len,next;}e[N*2];
struct data
{
    int x,y;

    bool operator < (const data &a) const
    {
        return y<a.y;
    }
};
set<data> w;

void addedge(int u,int v,int len)
{
    e[++cnt].to=v;e[cnt].len=len;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].len=len;e[cnt].next=last[v];last[v]=cnt;
}

void dfs(int x)
{
    dep[x]=dep[fa[x][0]]+1;dfn[x]=++tim;
    for (int i=1;i<=16;i++) fa[x][i]=fa[fa[x][i-1]][i-1];
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa[x][0]) continue;
        len[e[i].to]=len[x]+e[i].len;
        fa[e[i].to][0]=x;
        dfs(e[i].to);
    }
}

LL get_len(int x,int y)
{
    int x1=x,y1=y;
    if (dep[x]<dep[y]) swap(x,y);
    for (int i=16;i>=0;i--)
        if (dep[fa[x][i]]>=dep[y]) x=fa[x][i];
    if (x==y) return len[x1]+len[y1]-2*len[x];
    for (int i=16;i>=0;i--)
        if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return len[x1]+len[y1]-2*len[fa[x][0]];
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<n;i++)
    {
        int x,y,len;
        scanf("%d%d%d",&x,&y,&len);
        addedge(x,y,len);
    }
    dfs(1);
    LL ans=0;
    for (int i=1;i<=m;i++)
    {
        int x;
        scanf("%d",&x);
        if (trea[x])
        {
            trea[x]=0;
            data u;u.x=x;u.y=dfn[x];
            w.erase(u);
            if (!w.empty())
            {
                int p=0,q=0;
                set<data>::iterator it=w.upper_bound(u);
                if (it!=w.end())
                {
                    data v=*it;ans-=get_len(x,v.x);q=v.x;
                }
                if (it!=w.begin())
                {
                    it--;data v=*it;ans-=get_len(x,v.x);p=v.x;
                }
                if (p&&q) ans+=get_len(p,q);
            }
        }
        else
        {
            trea[x]=1;
            data u;u.x=x;u.y=dfn[x];
            if (!w.empty())
            {
                int p=0,q=0;
                set<data>::iterator it=w.upper_bound(u);
                if (it!=w.end())
                {
                    data v=*it;ans+=get_len(x,v.x);q=v.x;
                }
                if (it!=w.begin())
                {
                    it--;data v=*it;ans+=get_len(x,v.x);p=v.x;
                }
                if (p&&q) ans-=get_len(p,q);
            }
            w.insert(u);
        }
        LL f=0;
        if (!w.empty())
        {
            set<data>::iterator it=w.end();it--;
            data v=*w.begin(),g=*it;
            f=get_len(v.x,g.x);
        }
        printf("%lld\n",ans+f);
    }
    return 0;
}