P3047 [USACO12FEB]Nearby Cows G（树形dp）

最新推荐文章于 2024-01-22 15:42:59 发布

eva_can(not)survive

最新推荐文章于 2024-01-22 15:42:59 发布

阅读量345

点赞数

CC 4.0 BY-SA版权

分类专栏： DP 随学洛谷学习文章标签：算法

本文链接：https://blog.youkuaiyun.com/Ghostttttttiii/article/details/125628527

洛谷学习同时被 3 个专栏收录

202 篇文章

订阅专栏

71 篇文章

订阅专栏

随学

64 篇文章

订阅专栏

本文介绍了一个有趣的树形DP问题USACO NearbyCows，并提供了详细的解题思路及代码实现。通过两次DFS分别从根节点向下更新子节点和从子节点向上更新根节点的方式，高效地解决了计算特定距离内点权和的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

[USACO12FEB]Nearby Cows G - 洛谷https://www.luogu.com.cn/problem/P3047一个很有趣的树形dp

我们可以设状态f【i】【j】， i为当前根，j为距离为j时的点权和

首先我们可以取1为根跑一遍dfs，将以i为根的子树的点权和记录下即用儿子更新父亲，此时1肯定已经求完了，我们就可以再一次从1开始跑dfs用父亲去更新儿子。

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <cstdio>
#include <string>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <cstring>
#include <set>
#include <cmath>
#include <map>
#include <bitset>
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
const int MN = 65005;
const int MAXN = 1e6 + 10;
const int INF = 0x3f3f3f3f;
#define IOS ios::sync_with_stdio(false)
#define lowbit(x) ((x)&(-x))
using P = pair<int, int>;

int n, k;
int head[MAXN];
int ver[MAXN];
int nxt[MAXN];
int cost[MAXN];
int cnt;
int dep[MAXN];
int dp[MAXN][21];
void add(int x, int y) {
	ver[++cnt] = y;
	nxt[cnt] = head[x];
	head[x] = cnt;
}

void dfs1(int p, int fa) {
	for (int i = head[p]; i; i = nxt[i]) {
		int v = ver[i];
		if (v == fa)
			continue;
//		dep[v] = dep[p] + 1;
		dfs1(v, p);
		for (int j = 1; j <= k; j++) {
			dp[p][j] += dp[v][j - 1];
		}
	}
}

void dfs2(int p, int fa) {
	for (int i = head[p]; i; i = nxt[i]) {
		int v = ver[i];
		if (v == fa)
			continue;
		for (int j = k; j >= 2; j--) {
			dp[v][j] += dp[p][j - 1] - dp[v][j - 2];
		}
		dp[v][1] += cost[p];
		dfs2(v, p);
	}
}

int main() {
	scanf("%d %d", &n, &k);
	int x, y;
	for (int i = 1; i <= n - 1; i++) {
		scanf("%d %d", &x, &y);
		add(x, y);
		add(y, x);
	}
	for (int i = 1; i <= n; i++) {
		scanf("%d", cost + i);
		dp[i][0] = cost[i];
	}
	dfs1(1, 0);
	dfs2(1, 0);
	for (int i = 1; i <= n; i++) {
		int ans = 0;
		for (int j = 0; j <= k; j++) {
			ans += dp[i][j];
		}
		printf("%d\n", ans);
	}
	return 0;
}