【题解】CH#56C异象石 LCA+set

题目链接

描述

Adera是Microsoft应用商店中的一款解谜游戏。
异象石是进入Adera中异时空的引导物，在Adera的异时空中有一张地图。这张地图上有N个点，有N-1条双向边把它们连通起来。起初地图上没有任何异象石，在接下来的M个时刻中，每个时刻会发生以下三种类型的事件之一：

1. 地图的某个点上出现了异象石（已经出现的不会再次出现）；
2. 地图某个点上的异象石被摧毁（不会摧毁没有异象石的点）；
3. 向玩家询问使所有异象石所在的点连通的边集的总长度最小是多少。
   请你作为玩家回答这些问题。

输入格式

第一行有一个整数N，表示点的个数。
接下来N-1行每行三个整数x,y,z，表示点x和y之间有一条长度为z的双向边。
第N+1行有一个正整数M。
接下来M行每行是一个事件，事件是以下三种格式之一：

• x 表示点x上出现了异象石

• x 表示点x上的异象石被摧毁
  ? 表示询问使当前所有异象石所在的点连通所需的边集的总长度最小是多少。

输出格式

对于每个 ? 事件，输出一个整数表示答案。

样例输入

6
1 2 1
1 3 5
4 1 7
4 5 3
6 4 2
10
+3
+1
?
+6
?
+5
?
-6
-3
?

样例输出

5
14
17
10

数据范围与约定

对于30%的数据，1 ≤ n, m ≤ 1000。
对于另20%的数据，地图是一条链，或者一朵菊花。
对于100%的数据，1 ≤ n, m ≤ 10⁵, 1 ≤ x, y ≤ n, x ≠ y, 1 ≤ z ≤ 10⁹。

如果按照时间戳从小到大的顺序，把出现异象石的节点排成一圈（首位相接），并且累加相邻两个节点之间的路径长度，最后得到的结果恰好是所求答案的两倍。
我们可以用set按照时间戳递增的顺序维护出现异象石的节点序列，并用一个变量ans记录序列中相邻两个节点之间的路径长度之和。

#include<cstdio>
#include<set>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
typedef set<int>::iterator IT;
typedef long long ll;
const int N=1e5+10;
int fa[N][20],hd[N],tot,n,m,vis[N],dep[N],dfn[N],cnt,pos[N];
set<int>s;
ll dis[N][20],ans;
struct Edge{
	int v,w,nx;
}e[N<<1];
void addedge(int u,int v,int w)
{
	e[tot].v=v;
	e[tot].w=w;
	e[tot].nx=hd[u];
	hd[u]=tot++;
}
void bfs()
{
	queue<int>q;
	q.push(1);vis[1]=1;dep[1]=1;
	while(q.size())
	{
		int u=q.front();q.pop();
		for(int i=hd[u];~i;i=e[i].nx)
		{
			int v=e[i].v;
			if(vis[v])continue;
			vis[v]=1;
			dep[v]=dep[u]+1;
			fa[v][0]=u;
			dis[v][0]=e[i].w;
			for(int j=1;j<20;j++)
			{
				fa[v][j]=fa[fa[v][j-1]][j-1];
				dis[v][j]=dis[fa[v][j-1]][j-1]+dis[v][j-1];
			}
			q.push(v);
		}
	}
}
void dfs(int u)//打上时间戳 
{
	dfn[u]=++cnt;pos[cnt]=u;
	for(int i=hd[u];~i;i=e[i].nx)
	{
		int v=e[i].v;
		if(!dfn[v])dfs(v);
	}
}
ll lca(int u,int v)
{
	ll ret=0;
	if(dep[u]>dep[v])swap(u,v);
	for(int i=19;i>=0;i--)if(dep[fa[v][i]]>=dep[u])ret+=dis[v][i],v=fa[v][i];
	if(u==v)return ret;
	for(int i=19;i>=0;i--)
	if(fa[u][i]!=fa[v][i])ret+=dis[u][i]+dis[v][i],u=fa[u][i],v=fa[v][i];
	return ret+dis[u][0]+dis[v][0];
}
IT L(IT it)
{
	if(it==s.begin())return --s.end();
	return --it;
}
IT R(IT it)
{
	if(it==--s.end())return s.begin();
	return ++it;
}

int main()
{
	//freopen("in.txt","r",stdin);
    memset(hd,-1,sizeof(hd));
    scanf("%d",&n);
    int u,v,w;
    for(int i=1;i<n;i++)
    {
    	scanf("%d%d%d",&u,&v,&w);
    	addedge(u,v,w),addedge(v,u,w);
	}
	bfs();
	dfs(1);
	char com[5];
	scanf("%d",&m);
	IT it;
	for(int i=1;i<=m;i++)
	{
		scanf("%s",com);
		if(com[0]=='+')
		{
			scanf("%d",&u);
			if(s.size())
			{
				it=s.lower_bound(dfn[u]);
				if(it==s.end())it=s.begin();
				v=*L(it);
				ans+=lca(u,pos[v])+lca(u,pos[*it])-lca(pos[v],pos[*it]);
			}
			s.insert(dfn[u]);
		}
		else if(com[0]=='-')
		{
			scanf("%d",&u);
			it=s.find(dfn[u]);
			v=*L(it),it=R(it);
			ans-=lca(u,pos[v])+lca(u,pos[*it])-lca(pos[v],pos[*it]);
			s.erase(dfn[u]);
		}
		else printf("%lld\n",ans/2);
	}
	return 0;
}

总结

STL维护，较好