求助,样例错误
查看原帖
求助,样例错误
114914
一只书虫仔楼主2021/6/13 13:48
#include <bits/stdc++.h>

using namespace std;

struct edge {
	int u;
	int v;
} e[200086];

int n, k, cnt;
int head[200086];
int w[200086];
int dp[200086][21];

void AddEdge (int u, int v) {
	e[++cnt].u = v;
	e[cnt].v = head[u];
	head[u] = cnt;
}

void dfs (int cur, int fa) {
	for (int p = head[cur]; p; p = e[p].v) {
		if (e[p].u == fa) continue;
		dfs(e[p].u, cur);
		for (int i = 1; i <= k; i++)
			dp[cur][i] += dp[e[p].u][i - 1];
	}
}

void dfs2 (int cur, int fa) {
	for (int p = head[cur]; p; p = e[p].v) {
		if (e[p].u == fa) continue;
		dp[e[p].u][1] += dp[cur][0];
		for (int i = k; i > 1; i--)
			dp[e[p].u][i] -= dp[e[p].u][i - 2];
		for (int i = 1; i <= k; i++)
			dp[e[p].u][i] += dp[cur][i - 1];
		dfs(e[p].u, cur);
	}
}

int main () {
	scanf("%d%d", &n, &k);
	for (int i = 1, u, v; i < n; i++) {
		scanf("%d%d", &u, &v);
		AddEdge(u, v);
		AddEdge(v, u);
	}
	for (int i = 1; i <= n; i++) scanf("%d", &w[i]);
	for (int i = 1; i <= n; i++) dp[i][0] = w[i];
	dfs(1, -1);
	dfs2(1, -1);
	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;
}

样例输出:

15
41
15
4
9
6
2021/6/13 13:48
加载中...