第一句子网 - 唯美句子、句子迷、好句子大全
第一句子网 > Zoj 3201 Tree of Tree

Zoj 3201 Tree of Tree

时间:2022-07-12 18:39:32

相关推荐

Zoj 3201 Tree of Tree

树树

时间限制:1秒内存限制:32768KB

你给一个树的每个节点的权重,你需要找到这棵树的指定大小的最大子树。

树定义

树是其中不包含任何周期的连通图。

输入

有在输入多个测试用例。

每种情况下的第一行是两个整数N(1<=N<=100),K(1<=K<=N),其中N是该树的节点的数量,K是子树的大小,其次通过与N的非负整数,其中第k个整数表示第k个节点的权重的行。接下来的N-1行描述了树,每行有两个整数这意味着在这两个节点之间的边缘。以上所有的指标都是零基础,这是保证树的描述是正确的。

产量

一行与对于每种情况,这是最大的子树的总权重的单个整数。

样例输入

31

102030

01

02

32

102030

01

02

样例输出

30

40

树型DP+背包问题!

f[i][j]表示以i为根结点有j个结点子树的最大权值。

以下代码仅供参考!

#include<iostream>#include<cstdio>#include<cstring>#include<vector>using namespace std;#define MAX 105vector<int> adj[MAX];int f[MAX][MAX],tot[MAX],weight[MAX];int ans,K;int max(int a,int b){return a>b?a:b;}int DFS(int u,int father){tot[u]=1;int i,j,k,v;for(i=0;i<adj[u].size();i++){v=adj[u][i];if(v==father) // 如果又回到了父节点的情况continue;tot[u]+=DFS(v,u);}f[u][1]=weight[u];for(i=0;i<adj[u].size();i++){v=adj[u][i];if(v==father)continue;for(j=tot[u];j>=1;j--){for(k=0;k<j&&k<=tot[v];k++){f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]);}}}if(tot[u]>=K)ans=max(ans,f[u][K]);return tot[u];}int main(){int N,i,u,v;while(~scanf("%d%d",&N,&K)){for(i=0;i<N;i++){scanf("%d",&weight[i]);adj[i].clear();}for(i=1;i<N;i++){scanf("%d%d",&u,&v);adj[u].push_back(v);adj[v].push_back(u);}memset(f,0,sizeof(f));ans=0;DFS(1,-1);printf("%d\n",ans);}return 0;}

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。