您的位置:首页 > 其它

(树形DP)tree of tree (zju)

2014-12-17 10:26 447 查看
题意,一棵树中有N个结点(最多为100个结点)。每个结点有一个树值

求该树的子树结点最多为K个的所有权值和最大是多少。

思路:用dp[cur][K]记录包含当前结点与其子树所形成的子树的最大值。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
#include<string>
#include<vector>
#include<cmath>
#include<algorithm>

using namespace std;

const int N = 100;
int n,m;
vector<int> v
;
long long dp

,num
,w
;

void dfs(int cur,int pre)
{
//cout<<cur<<' '<<pre<<endl;
for (int i=2;i<=m;i++) dp[cur][i]=0;
dp[cur][1]=w[cur];
for (int i=0,len=v[cur].size();i<len;i++) if (v[cur][i]!=pre)
{
int son=v[cur][i];
//cout<<son<<endl;
dfs(son,cur);
for (int j=m-1;j;j--)
{
for (int k=j+1;k<=m;k++)
{
if (dp[son][k-j] == 0) break;
dp[cur][k]=max(dp[cur][k],dp[cur][j]+dp[son][k-j]);
}
}
}
}

int main()
{
freopen("in","r",stdin);
while (cin>>n>>m)
{
//	cout<<n<<' '<<m<<endl;
for (int i=0;i<n;i++)
{
scanf("%lld",w+i);
v[i].clear();
num[i]=0;
}
for (int i=1;i<n;i++)
{
int a,b;
scanf("%d%d",&a,&b);
v[a].push_back(b);
v[b].push_back(a);
}
dfs(0,-1);

long long  ans=0;
for (int i=0;i<n;i++) ans=max(ans,dp[i][m]);
printf("%lld\n",ans);

}

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: