zoj 3201 简单树形dp Tree of Tree

#include<iostream>
#include<cstdio>
/// 从 一棵树上选取k 个节点的子树的节点和的最大值
/// 状态转移方程 ： dp[u][j]=max(dp[u][tt]+dp[v][j-tt],dp[u][j]);
/// 表示从父节点选取tt个，从孩子节点选取j-tt个的最大值
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
int n,k,dp[102][102],sum[102];
vector<int>p[102];
void dfs(int u,int fa)
{
int i,j,tt, v;
dp[u][1]=sum[u];
for(i=0; i<p[u].size(); i++)
{
v=p[u][i];
if(v==fa)continue;
dfs(v,u);
for(j=k; j>=1; j--)
{
for(tt=1; tt<=j; tt++)
{
dp[u][j]=max(dp[u][tt]+dp[v][j-tt],dp[u][j]);
}
}
}
}
int main()
{
while(scanf("%d%d",&n,&k)!=EOF)
{
int i,j,u,v,ans=0;
for(i=0; i<=n; i++)
{
p[i].clear();
for(j=0; j<=k; j++)
dp[i][j]=0;
}
for(i=1; i<=n; i++)cin>>sum[i];
for(i=1; i<n; i++)
{
cin>>u>>v;
u++,v++;
p[u].push_back(v);
p[v].push_back(u);
}
dfs(1,-1);
for(i=1; i<=n; i++)
{
ans=max(ans,dp[i][k]);
}
cout<<ans<<endl;
}
return 0;
}

最后

以上就是虚拟小丸子最近收集整理的关于zoj 3201 简单树形dp Tree of Tree的全部内容，更多相关zoj内容请搜索靠谱客的其他文章。