洛谷P3047 附近的牛树形dp

最新推荐文章于 2023-05-16 19:52:12 发布

SigmaQuadrant

最新推荐文章于 2023-05-16 19:52:12 发布

阅读量147

点赞数

分类专栏：比赛题解

本文链接：https://blog.csdn.net/qq_40859782/article/details/101085479

版权

比赛题解专栏收录该内容

218 篇文章 0 订阅

订阅专栏

给出一个 $n\leq1e5$ 的树，每个点都有一个点权，求所有的点距离 $k$ 以内的所有的点的点权和。
不错的题。先简单想：如果是要求这个点向下所有距离 $k$ 以内的点权和。这个比较容易， $f_{u,0}=w_{u}$ ， $f_{u,i}=\sum\limits_{v \in son_{u}}f_{v,i-1}+f_{u,0}$ 。
然后我们还要求这个点向上的那一部分，这个部分要顺着根再推一次，因为对于根来说，向下的这一部分就是全部距离 $k$ 的答案。
我把两个部分分开计算，用 $f_{u,i,0}$ 表示 $u$ 这个点距离为 $i$ 只包含向下部分， $f_{u,i,1}$ 表示 $u$ 这个点距离为 $i$ 全部部分。
前者推导同上，后者推导 $f_{v,i,1}=f_{v,i,0}+f_{u,i-1,1}-f_{v,i-2,0}$ 。
也就是全部部分 $=$ 下半部分 $+$ 父亲节点的全部分 $-$ 重复部分。
直接转移。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=1e5+7;
vector<int> go[N];
int n,k; 
int f[N][21][2]; 
void dfs1(int u,int fa) {
	for(auto &v:go[u]) {
		if(v==fa) continue;
		dfs1(v,u);
		for(int i=1;i<=k;i++) 
			f[u][i][0]+=f[v][i-1][0];
	}
}
void dfs2(int u,int fa) {
	for(auto &v:go[u]) {
		if(v==fa) continue;
		for(int i=1;i<=k;i++) {
			if(i>=2) f[v][i][1]+=f[u][i-1][1]-f[v][i-2][0];
			else f[v][1][1]+=f[u][0][1];
		}
		dfs2(v,u);
	}
} 
int main() {
	scanf("%d%d",&n,&k);
	for(int i=1;i<n;i++) {
		int u,v;
		scanf("%d%d",&u,&v);
		go[u].push_back(v);
		go[v].push_back(u);
	}
	for(int i=1;i<=n;i++) {
		int x;
		scanf("%d",&x);
		for(int j=0;j<=k;j++) {
			f[i][j][0]=x;
			f[i][j][1]=x;
		}
	}
	dfs1(1,0);
	for(int i=1;i<=n;i++) {
		for(int j=1;j<=k;j++) {
			f[i][j][1]=f[i][j][0];
		}
	}
	dfs2(1,0);
	for(int i=1;i<=n;i++)
		printf("%d\n",f[i][k][1]);  
	return 0;
}