BZOJ 3991: [SDOI2015]寻宝游戏 虚树 set

3991: [SDOI2015]寻宝游戏

Time Limit: 40 Sec  Memory Limit: 128 MB
Submit: 1651  Solved: 810
[Submit][Status][Discuss]

Description

 小B最近正在玩一个寻宝游戏，这个游戏的地图中有N个村庄和N-1条道路，并且任何两个村庄之间有且仅有一条路径可达。游戏开始时，玩家可以任意选择一个村庄，瞬间转移到这个村庄，然后可以任意在地图的道路上行走，若走到某个村庄中有宝物，则视为找到该村庄内的宝物，直到找到所有宝物并返回到最初转移到的村庄为止。小B希望评测一下这个游戏的难度，因此他需要知道玩家找到所有宝物需要行走的最短路程。但是这个游戏中宝物经常变化，有时某个村庄中会突然出现宝物，有时某个村庄内的宝物会突然消失，因此小B需要不断地更新数据，但是小B太懒了，不愿意自己计算，因此他向你求助。为了简化问题，我们认为最开始时所有村庄内均没有宝物

Input

 第一行，两个整数N、M，其中M为宝物的变动次数。

接下来的N-1行，每行三个整数x、y、z，表示村庄x、y之间有一条长度为z的道路。

接下来的M行，每行一个整数t，表示一个宝物变动的操作。若该操作前村庄t内没有宝物，则操作后村庄内有宝物；若该操作前村庄t内有宝物，则操作后村庄内没有宝物。

Output

 M行，每行一个整数，其中第i行的整数表示第i次操作之后玩家找到所有宝物需要行走的最短路程。若只有一个村庄内有宝物，或者所有村庄内都没有宝物，则输出0。

Sample Input

4 5
1 2 30
2 3 50
2 4 60
2
3
4
2
1

Sample Output

0
100
220
220
280

HINT

 1<=N<=100000

1<=M<=100000

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

题目要求维护虚树的边权和

树有一个美好的性质

按dfn排序之后依次走到下一个点+dfn最大最小两点距离即是边权和

用set维护即可

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;

typedef long long ll;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
inline void print(ll x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=100100,inf=0X3f3f3f3f;

int last[N],ecnt;
struct EDGE{int to,nt,val;}e[N<<1];
inline void add(int u,int v,int val)
{e[++ecnt]=(EDGE){v,last[u],val};last[u]=ecnt;}

int dfn[N],pos[N],tim;
int dep[N],anc[N][20];
ll dis[N];

void dfs(int u)
{
	dfn[u]=++tim,pos[tim]=u;
	for(int i=1;(1<<i)<=dep[u];++i)
		anc[u][i]=anc[anc[u][i-1]][i-1];
	for(int i=last[u],v;i;i=e[i].nt)
		if((v=e[i].to)!=anc[u][0])
		{
			anc[v][0]=u;
			dep[v]=dep[u]+1;
			dis[v]=dis[u]+e[i].val;
			dfs(v);
		}
}

inline int getlca(int u,int v)
{
	if(dep[u]<dep[v]) swap(u,v);
	int len=dep[u]-dep[v];
	for(int i=0;(1<<i)<=len;++i)
		if((1<<i)&len)
			u=anc[u][i];
	if(u==v) return u;
	for(int i=17;~i;--i)
		if(anc[u][i]!=anc[v][i])
			u=anc[u][i],v=anc[v][i];
	return anc[u][0];
}

inline ll getdis(int u,int v)
{
	int lca(getlca(u,v));
	return dis[u]+dis[v]-(dis[lca]<<1);
}

bool in[N];

set<int> st;

int main()
{
	int n=read(),Q=read();
	register int i,u,v,val,opt,pre,nxt;
	for(i=1;i<n;++i)
		u=read(),v=read(),val=read(),
		add(u,v,val),add(v,u,val);
	dfs(1);
	st.insert(-inf),st.insert(inf);
	ll ans(0);
	while(Q--)
	{
		u=read(),in[u]^=1;
		in[u] ? (st.insert(dfn[u]),opt=1) : (st.erase(dfn[u]),opt=-1);
		pre=*--st.lower_bound(dfn[u]),nxt=*st.upper_bound(dfn[u]);
		if(pre!=-inf) ans+=opt*getdis(pos[pre],u);
		if(nxt!=inf) ans+=opt*getdis(pos[nxt],u);
		if(pre!=-inf && nxt!=inf) ans-=opt*getdis(pos[pre],pos[nxt]);
		if(st.size()>3)
			print(ans+getdis(pos[*st.upper_bound(-inf)],pos[*--st.lower_bound(inf)])),puts("");
		else print(ans),puts("");
	}
	return 0;
}