Description
网上出现了一种高科技产品——人品测试器。只要你把你的真实姓名输入进去,系统将自动输出你的人品指数。yzx不相信自己的人品为0。经过了许多研究后,yzx得出了一个更为科学的人品计算方法。这种方法的理论依据是一个非常重要的结论:人品具有遗传性。因此,一个人的人品完全由他的祖先决定。yzx提出的人品计算方法相当简单,只需要将测试对象的k个祖先的人品指数(可能为负数)加起来即可。选择哪k个祖先可以由测试者自己决定,但必须要满足这个要求:如果除自己的父母之外的某个祖先被选了,那么他的下一代必需要选(不允许跳过某一代选择更远的祖先,否则将失去遗传的意义)。
非常不幸的是,yzx测试了若干次,他的人品值仍然不能为一个正数。现在yzx需要你帮助他找到选择祖先的最优方案,使得他的人品值最大。
Input
第一行是两个用空格隔开的正整数n和k,其中n代表yzx已知的家谱中共有多少人(包括yzx本身在内),k的意义参见问题描述。
第二行有n-1个用空格隔开的整数(可能为负),这些数的绝对值在2^15以内。其中,第i个数表示编号为i+1的人的人品值。我们规定,编号为1的人是yzx。
接下来n行每行有两个用空格隔开的数,其中第i行的两个数分别表示第i个人的父亲和母亲的编号。如果某个人的父亲或母亲不在这个家谱内,则在表示他的父亲或母亲的编号时用0代替。
除yzx以外的所有人都是yzx的祖先,他们都会作为父亲或母亲被描述到。每个人都不可能同时作为多个人的父亲或者是母亲。
Output
一个整数,表示yzx能够得到的最大人品值。
Sample Input
6 3
-2 3 -2 3 -1
2 3
4 5
0 6
0 0
0 0
0 0
Sample Output
4
样例说明下图显示了输入样例所描述的家谱图。括号里的数表示的是该人的人品值。
4(-2) 5(3) 6(-1)
\ / /
\ / /
2(-2) 3(3)
\ /
\ /
1 <---yzx
显然,选择祖先2、3、5能使yzx的人品值达到最大。这个最大值为4,表示yzx能够得到的最大人品值。
Data Constraint
50%的数据,n<=10。
100%的数据,n<=100。
Source / Author: 常州高级中学 rpwt
题解:
树形dp
设f[i][j]表示选i并且选(j-1)个i的子节点的最大答案
f[i][j] = max(f[son[i][1]][k] + f[son[i][2]][j-1-k] + f[i][1]);
#include<bits/stdc++.h>
#define N 110
#define inf 2147483647
#define ll long long
#define point(a) multiset<a>::iterator
#define mod (ll)(500000)
#define mem(a,b) memset(a,b,sizeof (a))
#define open(x) freopen(x".in","r",stdin);freopen(x".out","w",stdout);
using namespace std;
int n,K,i,t,ans;
int f[N][N],son[N][2],a[N],num[N];
void dfs(int x)
{
if(!x)return ;
f[x][1]=a[x];
f[x][0]=0;
num[x]=1;
dfs(son[x][0]), dfs(son[x][1]);
num[x]+=num[son[x][0]] + num[son[x][1]];
int len=0 ;
if(x==1)len = min(K+1,num[x]) ; else len = min(K,num[x]);
for(int j=2;j<=len;j++)
{
f[x][j]=-inf;
for(int k=0;k<j;k++)if(k<=num[son[x][0]] && j-1-k<=num[son[x][1]])
f[x][j]= max(f[x][j] , f[x][1] + f[son[x][0]] [k] + f[son[x][1]] [j-1-k]);
}
}
int main()
{
open("moral");
scanf("%d%d",&n,&K);
for(i=1;i<n;i++)scanf("%d",&a[i+1]);
for(i=1;i<=n;i++) scanf("%d%d",&son[i][0],&son[i][1]);
dfs(1);
printf("%d",f[1][K+1]);
return 0;
}
时间复杂度:
O(n^3)
下面有一个O(n^2)算法
设f[i][j]为dfs序为i的点选i并且选(j-1)个i的子节点的最大答案
f[i][j] - > f[i+1][j+1]