1. 程式人生 > >洛谷P2014——選課

洛谷P2014——選課

pri scan %d print ext esp ref true printf

題目:https://www.luogu.org/problemnew/show/P2014

樹狀DP,註意枚舉當前子樹中選幾個時的邊界。

代碼如下:

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,k,s[305],ct,cnt[305],dp[305][305],size[305];
struct N{
	int to,next;
}edge[305];
void add(int x,int y)
{
	ct++;
	edge[ct].to=y;
	edge[ct].next=cnt[x];
	cnt[x]=ct;
}
void f(int nw)
{
	for(int i=cnt[nw];i;i=edge[i].next)
	{
		int u=edge[i].to;
		f(u);
		for(int j=min(size[nw]+size[u],m-1);j>=0;j--)//
		{
			for(int k=max(0,j-size[nw]);k<j&&k<=size[u];k++)//
				dp[nw][j]=max(dp[nw][j],dp[nw][j-k]+dp[u][k]);
		}
		size[nw]+=size[u];
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d",&k,&s[i]);
		add(k,i);
	}
	for(int i=1;i<=n;i++)size[i]=1,dp[i][1]=s[i];
	for(int i=cnt[0];i;i=edge[i].next)
	{
		int u=edge[i].to;
		f(u);
		for(int j=min(size[0]+size[u],m);j>=0;j--)
		{
			for(int k=max(0,j-size[0]);k<=size[u]&&k<=j;k++)
				dp[0][j]=max(dp[0][j],dp[0][j-k]+dp[u][k]);
		}
		size[0]+=size[u];
	}
	printf("%d",dp[0][m]);
	return 0;
}

洛谷P2014——選課