正题
题目大意
一棵树每个点有权值,选择kkk个点,要求选子节点之前必须要选父节点。求最大权值。
解题思路
树形背包限制一下必须选kkk个就好了
codecodecode
#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 110
using namespace std;
struct node{int to,next;
}a[N];
int n,k,tot,ls[N],f[N][N],w[N];
void addl(int x,int y){a[++tot].to=y;a[tot].next=ls[x];ls[x]=tot;
}
void dp(int x){if(x!=1) f[x][1]=w[x];else f[x][0]=0;for(int l=ls[x];l;l=a[l].next){int y=a[l].to;dp(y);for(int i=k;i>=1;i--)for(int j=1;j<=k;j++)if(i-j<0) break;else f[x][i]=max(f[x][i],f[x][i-j]+f[y][j]);}return;
}
int main(){memset(f,0xcf,sizeof(f));scanf("%d%d",&n,&k);for(int i=2;i<=n;i++)scanf("%d",&w[i]);for(int i=1;i<=n;i++){int x,y;scanf("%d%d",&x,&y);if(x) addl(i,x);if(y) addl(i,y);}dp(1);printf("%d",f[1][k]);
}