P3320 [SDOI2015]寻宝游戏

HyperlinkHyperlink

https://www.luogu.com.cn/problem/P3320


DescriptionDescription

给定一棵大小为nn的无根树,初始树上的节点都是白色的,有mm次操作,每次操作会将一个节点的颜色取反(黑变白,白变黑),问遍历完所有黑点的最短路径长度

数据范围:n105n\leq 10^5


SolutionSolution

显然从任意一个黑点出发的最短路径都是一定的,所以我们可以默认以1为根

我们可以发现,如果所有的路径都按统一的方式进行的话无疑路径是最短的,这个统一的方式显然可以选择欧拉序

那么我们只需要维护一个数据结构,支持插入,删除,查询前驱和后继

显然用setset

对于插入的每个点xx,找到它的前驱yy,后继zz(这里的前驱后继指的是在dfsdfs序中)

显然取反xx的贡献是dist(x,y)+dist(x,z)dist(y,z)dist(x,y)+dist(x,z)-dist(y,z)

由于是在树上,所以dist(x,y)=dis[x]+dis[y]2dis[lca(x,y)]dist(x,y)=dis[x]+dis[y]-2dis[lca(x,y)]

用倍增做lcalca,时间复杂度:O((n+m)logn)O((n+m)logn)


CodeCode

#include<set>
#include<queue>
#include<cmath>
#include<cstdio>
#include<vector>
#include<cctype>
#include<cstring>
#include<algorithm>
#define N 100010
#define LL long long
using namespace std;int n,m,root=1,x,y,f[N][17],dep[N],mdep,t,flag[N],tot,l[N],z,dfn[N],idf[N],cnt;
LL dis[N],ans;
bool vis[100001];
struct node{int next,to,w;}e[N*2];
inline void add(int u,int v,int w){e[++tot]=(node){l[u],v,w};l[u]=tot;return;}
set<int>s;
set<int>::iterator it;
inline long long read()
{
    char c;int d=1;long long f=0;
    while(c=getchar(),!isdigit(c))if(c==45)d=-1;f=(f<<3)+(f<<1)+c-48;
    while(c=getchar(),isdigit(c)) f=(f<<3)+(f<<1)+c-48;
    return d*f;
}
inline void dfs(int x)
{
	vis[x]=true;dfn[x]=++cnt;idf[cnt]=x;
	for(register int i=l[x];i;i=e[i].next)
	{
			int y=e[i].to;
			if(!vis[y])
			{
				vis[y]=true;
				dep[y]=dep[x]+1;
				dis[y]=dis[x]+e[i].w;
				mdep=max(mdep,dep[y]);
				f[y][0]=x;
				dfs(y);
			}
	}
	return;
}
inline int lca(int x,int y)
{
	if(dep[x]<dep[y]) swap(x,y);
	for(register int i=16;i>=0;i--) 
	if(dep[x]-(1<<i)>=dep[y]) x=f[x][i];
	for(register int i=16;i>=0;i--)
	if(f[x][i]!=f[y][i])
	{
		x=f[x][i];
		y=f[y][i];
	}
	if(x==y) return x;
	return f[x][0];
}
inline LL dist(int x,int y){return dis[x]+dis[y]-2*dis[lca(x,y)];}
signed main()
{
	n=read();m=read();flag[n]=-1;
	for(register int i=1;i<n;i++)
	{
		x=read();y=read();z=read();
		add(x,y,z);add(y,x,z);flag[i]=-1;
	}
	dfs(root);
	f[root][0]=-1;
	for(register int j=1;j<=floor(log(mdep)/log(2));j++)
	 for(register int i=1;i<=100000;i++)
	  if(f[i][j-1]>-1)f[i][j]=f[f[i][j-1]][j-1];else f[i][j]=-1;
	while(m--)
	{
		x=dfn[read()];
		if(flag[idf[x]]<0) s.insert(x);
		int y=idf[(it=s.lower_bound(x))==s.begin()?*--s.end():*--it];
		int z=idf[(it=s.upper_bound(x))==s.end()?*s.begin():*it];
		if(flag[idf[x]]>0) s.erase(x);
		x=idf[x];flag[x]*=-1;
		ans+=flag[x]*(dist(x,y)+dist(x,z)-dist(y,z));
		printf("%lld\n",ans);
	}
}
©️2020 CSDN 皮肤主题: 程序猿惹谁了 设计师: 上身试试 返回首页
实付0元
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值