洛谷 P3066 [USACO12DEC]Running Away From the Barn G(LCA + 二分 + 差分)

题目链接

刚看到这题以为是线段树合并,后来才发现边带权

我们考虑每个点对别人做出的贡献,也就是看他被多少个节点计入了答案
对于一个点 x x x,显然他只能对 x x x 到根的路径上的点做出贡献。由于这条路径上的点离 x x x 的距离满足单调性,所以受到 x x x 的贡献的节点一定是从 x x x 开始往上的连续一段节点。
那么,我们考虑 dfs 维护当前节点 x x x 到根的路径上的节点,并用 二分 + LCA 求出对哪些做出贡献,最后用树上差分维护每个节点的答案

这里贴的是没开 long long 的代码,因为开了 long long 的代码太丑了

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int Maxn=200010;
const int Maxm=Maxn<<1;
int c[Maxn],d[Maxn];
int f[Maxn][20];
int dis[Maxn],head[Maxn];
int s[Maxn],nxt[Maxm];
int to[Maxm],len[Maxm];
int n,m,edgecnt,cnt;
inline int read()
{
    
    
	int s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){
    
    if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0' && ch<='9')s=(s<<3)+(s<<1)+(ch^48),ch=getchar();
	return s*w;
}
inline void add(int x,int y,int c)
{
    
    
	++edgecnt;
	nxt[edgecnt]=head[x];
	to[edgecnt]=y;
	len[edgecnt]=c;
	head[x]=edgecnt;
}
int find(int x,int y)
{
    
    
	while(d[x]!=d[y])
	{
    
    
		if(d[x]<d[y])swap(x,y);
		for(int i=18;i>=0;--i)
		if(d[f[x][i]]>=d[y])x=f[x][i];
	}
	if(x==y)return x;
	for(int i=18;i>=0;--i)
	if(f[x][i]!=f[y][i])
	x=f[x][i],y=f[y][i];
	return f[x][0];
}
int dist(int x,int y)
{
    
    
	int pos=find(x,y);
	return dis[x]+dis[y]-(dis[pos]<<1);
}
void dfs(int x,int fa,int pre)
{
    
    
	d[x]=d[fa]+1;
	dis[x]=dis[fa]+pre;
	f[x][0]=fa;
	for(int i=1;i<=18;++i)
	f[x][i]=f[f[x][i-1]][i-1];
	c[++cnt]=x;
	int l=1,r=cnt;
	while(l<r)
	{
    
    
		int mid=(l+r)>>1;
		if(dist(x,c[mid])>m)l=mid+1;
		else r=mid;
	}
	++s[x],--s[f[c[l]][0]];
	for(int i=head[x];i;i=nxt[i])
	{
    
    
		int y=to[i];
		if(y==fa)continue;
		dfs(y,x,len[i]);
	}
	--cnt;
}
void work(int x,int fa)
{
    
    
	for(int i=head[x];i;i=nxt[i])
	{
    
    
		int y=to[i];
		if(y==fa)continue;
		work(y,x),s[x]+=s[y];
	}
}
int main()
{
    
    
//	freopen("in.txt","r",stdin);
	n=read(),m=read();
	for(int i=2;i<=n;++i)
	{
    
    
		int x=read(),val=read();
		add(x,i,val),add(i,x,val);
	}
	dfs(1,0,0);
	work(1,0);
	for(int i=1;i<=n;++i)
	printf("%lld\n",s[i]);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/Brian_Pan_/article/details/110928431