洛谷P2014 [CTSC1997]选课（树形dp+背包）

最新推荐文章于 2024-03-28 23:45:00 发布

Orz_Sponge_Bob

最新推荐文章于 2024-03-28 23:45:00 发布

阅读量542

点赞数

本文链接：https://blog.csdn.net/weixin_45948940/article/details/119871748

版权

这篇博客探讨了一种使用树形动态规划（DP）解决森林中节点最大分数问题的方法。作者首先尝试使用网络流但发现模型存在环，于是转向树形DP。通过将森林转化为树，并定义dp状态表示在节点i选择一定数量儿子所能获得的最大分数，实现了01背包的经典转化。在DFS过程中处理子节点并进行背包转移，最终得出最大贡献。博客提供了完整的C++代码实现。

摘要由CSDN通过智能技术生成

题目链接
在这里插入图片描述
前言：怎么这树形dp入门题都这么难QAQ
题目分析：
写了发网络流，挂了，看了看数据发现我以虚拟节点建立的模型成环了emmm然后难搞了。还是得树形dp。
首先是这个森林，为了方便我们先把他们汇聚到树上，用虚拟节点的形式。
我们定义dp[i][j][k]表示节点i 的前j个儿子中选择k个所能获得的最大分数。
emmm这里就很01背包了。
压缩一维变为dp[i][j] 表示在节点i，限度为j的情况下能获得的最大贡献。
dfs过程中先处理儿子节点。然后再对该根背包转移。背包总容量为M，物品的V不确定的情况下我们靠枚举来确定它。然后就是经典01背包了QAQ

 #include<bits/stdc++.h>
#include<stdlib.h>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<queue>
#include<time.h>
#include <cstdio>
#include <iostream>
#include <vector>
#define ll long long
#define int long long
#define inf 0x3f3f3f3f
#define mods 1000000007
#define modd 998244353
#define PI acos(-1)
#define fi first
#define se second
#define lowbit(x) (x&(-x))
#define mp make_pair
#define pb push_back
#define si size()
#define E exp(1.0)
#define fixed cout.setf(ios::fixed)
#define fixeds(x) setprecision(x)
#define IOS ios::sync_with_stdio(false);cin.tie(0)
 using namespace std;
 ll gcd(ll a,ll b){if(a<0)a=-a;if(b<0)b=-b;return b==0?a:gcd(b,a%b);}
template<typename T>void read(T &res){bool flag=false;char ch;while(!isdigit(ch=getchar()))(ch=='-')&&(flag=true);
for(res=ch-48;isdigit(ch=getchar());res=(res<<1)+(res<<3)+ch - 48);flag&&(res=-res);}
ll lcm(ll a,ll b){return a*b/gcd(a,b);}
ll qp(ll a,ll b,ll mod){ll ans=1;if(b==0){return ans%mod;}while(b){if(b%2==1){b--;ans=ans*a%mod;}a=a*a%mod;b=b/2;}return ans%mod;}//快速幂%
ll qpn(ll a,ll b, ll p){ll ans = 1;a%=p;while(b){if(b&1){ans = (ans*a)%p;--b;}a =(a*a)%p;b >>= 1;}return ans%p;}//逆元   (分子*qp(分母,mod-2,mod))%mod;


vector<ll>G[310];
ll a[310];
ll dp[333][333];
ll m;
void dfs(ll rt)
{

    for(int i=0; i<G[rt].size(); i++)
    {
        ll v=G[rt][i];
        dfs(v);
    }

    for(int i=0; i<G[rt].size(); i++)
    {

        for(int j=m; j>=1; j--)
        {

            for(int k=0; k<j; k++)
            {

                dp[rt][j]=max(dp[rt][j],dp[rt][j-k]+dp[G[rt][i]][k]);


            }


        }



    }


}
signed main()
{
    ll n;
    read(n);
    read(m);
    m++;
    for(int i=1; i<=n; i++)
    {

        ll x,y;
        read(x);
        read(a[i]);
        G[x].push_back(i);
        dp[i][1]=a[i];
    }
    dfs(0*1ll);
    printf("%lld",dp[0][m]);
}