题目:
https://www.luogu.org/problem/show?pid=2014
dp[i][j]:以i为根选j个物品(一定包括根);
因为是森林,所以要建一个超级原点,把他们转化为一棵树;
怎么建?
因为n>=1;
所以,所有森林里的树的根的父亲默认为0,
so 不用建~~~~~(我纠结了很久QAQ)
先拓扑排序再DP(树形套路???)
边界:
所有dp[i][j]初始化为w[i],以保证以i为根i一定被选上;
这样就变成了一个在拓扑序上做的普通01背包问题~~
关键是转化,转化后就挺简单了;
感觉老师讲复杂了(好吧,还是我太弱);
方程:
dp[fa[x]][i]=max(dp[fa[x]][i-j]+dp[x][j],dp[fa[x]][i]);
注意三个问题:
1.m需要+1,因为0本身需要选上;
2.倒序更新,去掉后效性;
3.考虑dp[fa[x]][i-j]+dp[x][j]这句,fa[x]的i-j是否已包含x?
答案是否的,因为我们用要x更新它父亲,此前x与它父亲并无关联(因为是在拓扑序上DP),所以dp[fa[x]][i-j]中的i-j并不包含x,明白这点,这个题就很好理解了;
如果没有发现这个问题,说明还没有明白方程来龙去脉,再好好想想;
收获:
边界的巧妙设置~~;
转化的魅力~~~
拓扑排序消除后效性~~~;
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int MAXN=20001;
int w[MAXN],dp[301][301],fa[MAXN],son[MAXN];
int n,m;
queue<int>q;
void top_sort()
{
while(!q.empty())
{
int x=q.front();
q.pop();
son[fa[x]]--;
if(!son[fa[x]]) q.push(fa[x]);
for(int i=m;i>=1;i--)
for(int j=i-1;j>=1;j--)
dp[fa[x]][i]=max(dp[fa[x]][i-j]+dp[x][j],dp[fa[x]][i]);
}
return;
}
void solve()
{
scanf("%d%d",&n,&m),m++;
for(int i=1;i<=n;i++)
{
scanf("%d%d",&fa[i],&w[i]),son[fa[i]]++;
for(int j=1;j<=m;j++) dp[i][j]=w[i];
}
fa[0]=-1;
for(int i=1;i<=n;i++) if(!son[i]) q.push(i);
top_sort();
cout<<dp[0][m];
return;
}
int main()
{
solve();
return 0;
}