洛谷 p2014 [CTSC2001] 选课 树形背包DP

题目:
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
#include
#include
#include
#include
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<0][m];
    return;
}
int main()
{
    solve();
    return 0;
} 

你可能感兴趣的:(noip,dp)