URAL1018. Binary Apple Tree

链接

简单树形DP

 #include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stdlib.h>
using namespace std;
#define INF 0xfffffff
int dp[][];
int n,q,ch[][];
int p[][];
int dfs(int pre,int u,int s)
{
int i;
if(dp[u][s]!=-)
return dp[u][s];
int o = ,cc[];
for(i = ; i <= n ; i++)
{
if(i==pre)
continue;
if(p[u][i])
{
o++;
cc[o] = i;
}
}
if(s==||o==)
return dp[u][s] = ;
if(o==&&s>)
return -INF;
if(o==)
{
dp[u][s] = p[u][cc[]]+dfs(u,cc[],s-);
return dp[u][s];
}
dp[u][s] = max(p[u][cc[]]+dfs(u,cc[],s-),p[u][cc[]]+dfs(u,cc[],s-));
for(i = ; i < s ; i++)
{
dp[u][s] = max(p[u][cc[]]+dfs(u,cc[],i-)+p[u][cc[]]+dfs(u,cc[],s-i-),dp[u][s]);
}
return dp[u][s];
}
int main()
{
int i,u,v,w;
memset(dp,-,sizeof(dp));
scanf("%d%d",&n,&q);
for(i = ; i < n ; i++)
{
scanf("%d%d%d",&u,&v,&w);
p[u][v] = w;
p[v][u] = w;
}
int ans = dfs(-,,q);
printf("%d\n",ans);
return ;
}
上一篇:df、du、fdisk:Linux磁盘管理三板斧的使用心得(转载)


下一篇:Django REST framework 第三章 CBV