1. 程式人生 > >【洛谷P2014】選課【樹形DP】【揹包】

【洛谷P2014】選課【樹形DP】【揹包】

題目大意:

題目連結:https://www.luogu.org/problemnew/show/P2014
n n 門功課,一些功課有先修課。每門功課都有學分。求選出 m m 門功課能獲得的最大學分。


思路:

樹形DP+揹包。
很明顯,這道題肯定是設 f [ u ] [ j ] f[u][j]

表示以 u u 為根的子樹選出 j j 門課程學習能獲得的最大學分。
那麼對於 u u
的任意一棵子樹 v v ,我們設它有 s s 個結點,那麼我們就可以在這棵子樹中選擇任意 k [ 0 s ] k\in[0\sim s] 個結點來轉移。
那麼方程很明顯就是
f [ u ] [ j ] = m a x ( f [ u ] [ j ] , f [ u ] [ j k 1 ] + f [ v ] [ k ] ) f[u][j]=max(f[u][j],f[u][j-k-1]+f[v][k])
為什麼 j k j-k 之後還要減 1 1 呢?
因為要選擇 u u 的子樹,那麼 u u 是肯定得選的,所以在轉移的時候可以忽略掉這一個結點
那麼很明顯,最終答案就是 f [ 0 ] [ m ] f[0][m]


程式碼:

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 400
using namespace std;

int n,m,f[N][N],head[N],tot;

struct edge
{
	int next,to,s;
}e[N]; 

void add(int from,int to,int s)
{
	e[++tot].to=to;
	e[tot].s=s;
	e[tot].next=head[from];
	head[from]=tot;
}

int dp(int x)
{
	int sum=1;  //以x為跟的節點個數
	for (int i=head[x];~i;i=e[i].next)
	{
		int sonsum=dp(e[i].to);  //子樹節點個數
		sum=sum+sonsum;
		for (int j=sum-1;j>=0;j--)  //揹包,要降序
		 for (int k=0;k<sonsum;k++)
		  if (j-k>=1)
		   f[x][j]=max(f[x][j],f[x][j-k-1]+f[e[i].to][k]);
	}
	return sum;
}

int main()
{
	memset(head,-1,sizeof(head));
	scanf("%d%d",&n,&m);
	int x,y;
	for (int i=1;i<=n;i++)
	{
		scanf("%d%d",&x,&y);
		f[i][0]=y;
		add(x,i,y);
	}
	dp(0);
	printf("%d\n",f[0][m]);
	return 0;
}