http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3201
题目大意:给出一个树,每个点有一个值,选择k个点要求值最大,k各点连成一颗子树,求最大值
代码:
#include <math.h>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define MAXN 105
#define INF 0x7fffffff
#define eps 1e-10
#define MAX(x,y) ((x)>(y)? (x):(y))
#define MIN(x,y) ((x)<(y)? (x):(y))
#define MEM(a) (memset((a),0,sizeof(a)))
#define FRE freopen("input.txt","r",stdin)
using namespace std;
int n,m;
int val[MAXN],head[MAXN],vis[MAXN],dp[MAXN][MAXN];
struct edge
{
int u,v,next;
}e[2*MAXN];
void dfs(int u)
{
vis[u]=1;
for(int i=head[u];i!=-1;i=e[i].next)
{
int v=e[i].v;
if(!vis[v])
{
dfs(v);
for(int j=m;j>0;j--)
for(int k=0;k<=m-j;k++)
dp[u][j+k]=MAX(dp[u][j+k],dp[u][j]+dp[v][k]);
}
}
return;
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(head,-1,sizeof(head));
MEM(dp);
for(int i=0;i<n;i++)
scanf("%d",&val[i]);
int cnt=0;
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
e[cnt].u=u; e[cnt].v=v; e[cnt].next=head[u]; head[u]=cnt++;
e[cnt].u=v; e[cnt].v=u; e[cnt].next=head[v]; head[v]=cnt++;
}
for(int i=0;i<n;i++)
dp[i][1]=val[i];
int ans=-1;
MEM(vis);
dfs(0);
for(int i=0;i<n;i++)
{
ans=MAX(ans,dp[i][m]);
}
printf("%d\n",ans);
}
return 0;
}