P3320 [SDOI2015]寻宝游戏 题解

const int N=1e5+13;
struct Edge{int v,w,nxt;}e[N<<1];
int n,m,h[N],tot;
ll dis[N];
inline void add_edge(int u,int v,int w){e[++tot]=(Edge){v,w,h[u]};h[u]=tot;}
int fa[N],dep[N],siz[N],son[N],top[N],id[N],dfs_clock;
void dfs1(int u,int f,int deep){
	dep[u]=deep,fa[u]=f,siz[u]=1;
	int maxson=0;
	for(int i=h[u];i;i=e[i].nxt){
		int v=e[i].v;if(v==f) continue;
		dis[v]=dis[u]+e[i].w;
		dfs1(v,u,deep+1);
		siz[u]+=siz[v];
		if(siz[v]>maxson) maxson=siz[v],son[u]=v;
	}
}
void dfs2(int u,int topf){
	top[u]=topf,id[u]=++dfs_clock;
	if(!son[u]) return;
	dfs2(son[u],topf);
	for(int i=h[u];i;i=e[i].nxt){
		int v=e[i].v;
		if(v!=fa[u]&&v!=son[u]) dfs2(v,v);
	}
}
inline int lca(int u,int v){
	while(top[u]!=top[v]){
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		u=fa[top[u]];
	}
	return id[u]<id[v]?u:v;
}
inline ll dist(int u,int v){return dis[u]+dis[v]-2*dis[lca(u,v)];}
struct Node{
	int x;
	bool operator <(const Node &a)const{return id[x]<id[a.x];}
};
std::set<Node> t;
std::set<Node>::iterator it;
int main(){
//file();
	read(n),read(m);
	for(int i=1;i<n;++i){
		int u,v,w;read(u),read(v),read(w);
		add_edge(u,v,w),add_edge(v,u,w);
	}
	dfs1(1,0,0),dfs2(1,1);
	ll sum=0;
	while(m--){
		int x;read(x);
		if(t.find((Node){x})!=t.end()){//remove
			it=t.find((Node){x});
			int pre=0,nxt=0;
			if(it!=t.begin()) pre=(--it)->x,it=t.find((Node){x});
			if((++it)!=t.end()) nxt=it->x;
			if(pre) sum-=dist(pre,x);if(nxt) sum-=dist(nxt,x);if(pre&&nxt) sum+=dist(pre,nxt);
			t.erase((Node){x});
		}
		else{//insert
			t.insert((Node){x});
			it=t.find((Node){x});
			int pre=0,nxt=0;
			if(it!=t.begin()) pre=(--it)->x,it=t.find((Node){x});
			if((++it)!=t.end()) nxt=it->x;
			if(pre) sum+=dist(pre,x);if(nxt) sum+=dist(nxt,x);if(pre&&nxt) sum-=dist(pre,nxt);
		}
		it=t.end();--it;
		if(it!=t.begin()){
			int u=it->x;it=t.begin();int v=it->x;
			println(sum+dist(u,v));
		}
		else println(sum);
	}
	return 0;
}