P2014 选课

这是树上背包型问题

$f[i][j]$表示对于以$i$为根的子树中选$j$个获得的学分

则有$f[i][j]=min(f[i][j-k-1]+f[v][k]+w[v])$

之所以$j-k-1$是因为子树中肯定要取$v$,对于每一个子树,都进行这样的处理,就可以得出答案

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#include<bits/stdc++.h>
using namespace std;
const int maxn=1000;
int n,m,w[maxn],f[maxn][maxn];
vector<int> son[maxn];
void dfs(int u)
{
	for(int i=0;i<son[u].size();i++)
	{
		dfs(son[u][i]);
		for(int j=m;j>=0;j--)
			for(int k=j-1;k>=0;k--)
			f[u][j]=max(f[u][j-k-1]+f[son[u][i]][k]+w[son[u][i]],f[u][j]);
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) 
	{
		int x,y;
		scanf("%d%d",&x,&y);
		son[x].push_back(i);
		w[i]=y;
	}
	dfs(0);
	printf("%d\n",f[0][m]);
	return 0;
}