洛谷P3320 [SDOI2015]寻宝游戏(虚树+set找前驱后继)

题目

n(n<=1e5) 个节点的树，边有边权w(w<=1e9)。

每个点可能是关键点，

m(m<=1e5)次操作，每次将已经是关键点的改为否，反之改为是。

动态维护，从任意一点出发，遍历所有关键点后回到该点，所经历的最小权值和。

思路来源

https://www.luogu.com.cn/blog/PinkRabbit/solution-p3320

题解

发现应该从关键点出发更优，考虑所有关键点的这棵虚树，

应该按dfs序遍历这棵树所得代价最小，且为虚树上所有边长的二倍，

于是，预处理dfs序，

插入一个虚树点x的时候，就考虑改变其与dfs序前驱点y和后继点z对答案带来的贡献，

插入x时，显然答案加上d(y,x)+d(x,z)-d(y,z)，

删除同理

心得

原来set可以支持找前驱和后继…

之前一直以为要手写的平衡树才行…

粉兔讨论前驱后继的代码好简洁

代码

#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define fi first
#define se second
typedef pair<int,int> P;
typedef long long ll;
const int N=1e5+10;
vector<P>e[N];
int n,m,x,y,z,dep[N],f[N][20],dfn[N],id[N],c;
ll d[N],ans;
set<int>now;
set<int>::iterator it;
bool vis[N];
void dfs(int u,int fa){
    dfn[++c]=u;
    id[u]=c;
    for(auto &x:e[u]){
        int v=x.fi,w=x.se;
        if(v==fa)continue;
        dep[v]=dep[u]+1;
        d[v]=d[u]+w;
        f[v][0]=u;
        dfs(v,u);
    }
}
int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    int z=dep[x]-dep[y];
    for(int j=19;j>=0;--j){
        if(z>>j&1){
            x=f[x][j];
        }
    }
    if(x==y)return x;
    for(int j=19;j>=0;--j){
        if(f[x][j]!=f[y][j]){
            x=f[x][j];
            y=f[y][j];
        }
    }
    return f[x][0];
}
ll g(int x,int y){
    //printf("x:%d y:%d lca:%d\n",x,y,lca(x,y));
    return d[x]+d[y]-2*d[lca(x,y)];
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;++i){
        scanf("%d%d%d",&x,&y,&z);
        e[x].pb(P(y,z));
        e[y].pb(P(x,z));
    }
    dfs(1,-1);
    for(int j=1;j<20;++j){
        for(int i=1;i<=n;++i){
            f[i][j]=f[f[i][j-1]][j-1];
        }
    }
    for(int i=1;i<=m;++i){
        scanf("%d",&x);
        int v=id[x];
        if(!vis[x])now.insert(v);
        //begin()首元素 end指向末元素的下一位置
        y=dfn[((it=now.lower_bound(v))==now.begin()? *(--now.end()):*(--it))];
        z=dfn[(it=now.upper_bound(v))==now.end()? *now.begin():*it];
        if(vis[x])now.erase(v);
        ll sum=g(y,x)+g(x,z)-g(y,z);
        if(!vis[x])vis[x]=1,ans+=sum;
        else vis[x]=0,ans-=sum;
        printf("%lld\n",ans);
    }
    return 0;
}