P2014 [CTSC1997]选课

题目

题目

思路

树形dp典型题，设f[i][j]为从第i节课开始报j节课的最优值。
$f_{i,j}=max(f_{i,k}+f_{to(为i子节点),j-k})，M+1>=j>=1,0<=k<j$
code：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<cmath>
#include<queue>
#include<deque>
using namespace std;
long long n,q,head[1001],b[1001];
struct fP{
	long long x,to,nxt;
} a[4001];
long long sie[1001],dp[1001][1001],mx;
void dfs(long long x,long long fa)
{
	sie[x]=1;
	dp[x][1]=b[x];
	for (long long i=head[x];i;i=a[i].nxt)
	{
		dfs(a[i].to,x);
		sie[x]+=sie[a[i].to];
		for (long long j=q+1;j;j--) for (long long k=j-1;k>=0;k--)
		{
			dp[x][j]=max(dp[x][j],dp[x][j-k]+dp[a[i].to][k]);
			mx=max(dp[x][j],mx);
		}
	}
	return;
}
int main()
{
	scanf("%lld%lld",&n,&q);
	for (int i=1;i<=n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		b[i]=y;
		a[i].to=i,a[i].x=x,a[i].nxt=head[x],head[x]=i;
	}
	dfs(0,-1);
	cout<<mx;
    return 0;
}