样例没过+20pts求助(或善良,或邪恶,或不善亦不邪,都来帮我调调吧!)
查看原帖
样例没过+20pts求助(或善良,或邪恶,或不善亦不邪,都来帮我调调吧!)
658786
STUDENT00楼主2022/11/14 19:16

马风很好,代码如下:

#include<bits/stdc++.h>
using namespace std;
int n,k,dp[100001][21],ans[100001][21],c[100001],fa[100001];
vector<int> w[100001],v[100001];
bool vis[100001];
void build(int now){
	vis[now]=1;
	for(int i=0;i<w[now].size();i++){
		int t=w[now][i];
		if(!vis[t]){
			vis[t]=1;
			v[now].push_back(t);
			fa[t]=now;
			build(t);
		}
	} 
}
void dfs(int now){
	for(int i=0;i<v[now].size();i++) dfs(v[now][i]);
	dp[now][0]=c[now];
	for(int i=0;i<v[now].size();i++){
		int t=v[now][i];
		for(int j=1;j<=k;j++) dp[now][j]+=dp[t][j-1];
	}
}
void dfs2(int now){
    ans[now][0]=c[now];
    for(int i=1;i<=k;i++){
        if(fa[now]){
            if(i==1) ans[now][i]=dp[now][i]+c[fa[i]];
            else ans[now][i]=ans[fa[now]][i-1]-dp[now][i-2]+dp[now][i];
        }else ans[now][i]=dp[now][i];
    }
    for(int i=0;i<v[now].size();i++) dfs2(v[now][i]);
}
int main(){
	scanf("%d%d",&n,&k);
	for(int i=1;i<n;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		w[u].push_back(v);
		w[v].push_back(u);
	}
	for(int i=1;i<=n;i++) scanf("%d",&c[i]);
	build(1);
	dfs(1);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=k;j++) dp[i][j]+=dp[i][j-1];
	}
    dfs2(1);
	for(int i=1;i<=n;i++) printf("%d\n",ans[i][k]);
	return 0;
}
2022/11/14 19:16
加载中...