LibreOJ 10132. 异象石

链接

https://loj.ac/p/10132

题意

树形图上有三种操作:

操作 1:某个点被标记

操作 2:某个点被取消标记

操作 3:求所有被标记的点连通的边集的最小值

总共有 $m$ 个操作

思路

将所有的点按照dfs序排序,易发现相邻两点的路径长度之和(首尾相连)恰好是答案的两倍

用 $set$ 按照dfs序大小维护被标记的点

标记点 $x$ :$ans+=path[l][x]+path[x][r]-path[l][r]$

取消点 $x$ :$ans+=-path[l][x]-path[x][r]+path[l][r]$

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+5;
int n,m;
int cnt,to[N*2],val[N*2],nxt[N*2],head[N];
int fa[N],depth[N],sze[N],top[N],son[N],tot,dfn[N];
ll dist[N],ans;
set<pair<int,int> >s;
void addedge(int u,int v,int w) {
cnt++;
to[cnt]=v;
val[cnt]=w;
nxt[cnt]=head[u];
head[u]=cnt;
}
void dfs1(int u,int t) {
dfn[u]=++tot;
depth[u]=t;
sze[u]=1;
for(int i=head[u];i;i=nxt[i]) {
int v=to[i];
if(v==fa[u]) continue;
fa[v]=u;
dist[v]=dist[u]+val[i];
dfs1(v,t+1);
sze[u]+=sze[v];
if(sze[v]>sze[son[u]]) son[u]=v;
}
}
void dfs2(int u,int t) {
top[u]=t;
if(!son[u]) return;
dfs2(son[u],t);
for(int i=head[u];i;i=nxt[i]) {
int v=to[i];
if(v==son[u]||v==fa[u]) continue;
dfs2(v,v);
}
}
int lca(int a,int b) {
while(top[a]!=top[b]) {
if(depth[top[a]]>depth[top[b]]) a=fa[top[a]];
else b=fa[top[b]];
}
return depth[a]<depth[b]?a:b;
}
ll path(int a,int b) {
return dist[a]+dist[b]-2*dist[lca(a,b)];
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
cin>>n;
for(int i=1;i<n;i++) {
int u,v,w;
cin>>u>>v>>w;
addedge(u,v,w);
addedge(v,u,w);
}
dfs1(1,1);
dfs2(1,1);
cin>>m;
for(int i=1;i<=m;i++) {
char c;
cin>>c;
if(c=='+') {
int x;
cin>>x;
s.insert({dfn[x],x});
if(s.size()==1) continue;
auto l=s.lower_bound({dfn[x],x});
auto r=s.upper_bound({dfn[x],x});
if(l==s.begin()) l=--s.end();
else l--;
if(r==s.end()) r=s.begin();
int lnode=(*l).second,rnode=(*r).second;
ans+=path(lnode,x)+path(x,rnode)-path(lnode,rnode);
}
else if(c=='-') {
int x;
cin>>x;
s.erase({dfn[x],x});
if(s.size()==0) continue;
auto l=s.lower_bound({dfn[x],x});
auto r=s.upper_bound({dfn[x],x});
if(l==s.begin()) l=--s.end();
else l--;
if(r==s.end()) r=s.begin();
int lnode=(*l).second,rnode=(*r).second;
ans+=-path(lnode,x)-path(x,rnode)+path(lnode,rnode);
}
else cout<<ans/2<<endl;
}
return 0;
}