【洛谷P2014】選課【樹形DP】【揹包】
阿新 • • 發佈:2018-11-10
題目大意:
題目連結:https://www.luogu.org/problemnew/show/P2014
有
門功課,一些功課有先修課。每門功課都有學分。求選出
門功課能獲得的最大學分。
思路:
樹形DP+揹包。
很明顯,這道題肯定是設
表示以
為根的子樹選出
門課程學習能獲得的最大學分。
那麼對於
的任意一棵子樹
,我們設它有
個結點,那麼我們就可以在這棵子樹中選擇任意
個結點來轉移。
那麼方程很明顯就是
為什麼
之後還要減
呢?
因為要選擇
的子樹,那麼
是肯定得選的,所以在轉移的時候可以忽略掉這一個結點。
那麼很明顯,最終答案就是
。
程式碼:
#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;
}