洛谷P2015 二叉苹果树&&洛谷P2014 选课 树形dp

参考博客：https://blog.csdn.net/kkkksc03/article/details/83018645

P2015 二叉苹果树

思路：dfs+回溯+dp

AC代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<string>
#include<queue>
#include<vector>
#include<cmath>
#include<stack>
#include<map>
#include<set>
using namespace std;
#define LL long long
const int MOD=100000007;
const int inf=0x3f3f3f3f;
const LL inff=0x3f3f3f3f3f3f3f3f;
const LL N=300005;
const LL M=5005;
#define MEF(x) memset(x,-1,sizeof(x))
#define ME0(x) memset(x,0,sizeof(x))
#define MEI(x) memset(x,inf,sizeof(x))
struct Edge
{
    int u,v,w,next;
}edge[205];
int cnt=0,first[105];
void init()
{
    cnt=0;
    MEF(first);
}
void add(int u,int v,int w)
{
    edge[cnt].u=u;
    edge[cnt].v=v;
    edge[cnt].w=w;
    edge[cnt].next=first[u];
    first[u]=cnt++;
}
int dp[105][105];//dp[i][j]表示第i号节点保留j个树枝剩下的的苹果的最大个数
void DP(int x,int fa,int q)
{
    for(int i=first[x];i!=-1;i=edge[i].next)
    {
        int lv=edge[i].v;
        if(lv==fa)
        {
            continue;
        }
        dp[lv][1]=edge[i].w;
        DP(lv,x,q);
        for(int j=q;j>=1;j--)
        {
            for(int k=0;k<=j;k++)
            {
                if((j!=1&&j!=k)||x==1)//当该节点不是根节点时，j个节点中一定要有父亲节点
                {
                    dp[x][j]=max(dp[x][j],dp[lv][k]+dp[x][j-k]);//枚举中间节点遍历左右儿子保留不同数量时的情况
                }
            }
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);
    int n,q;
    cin>>n>>q;
    init();
    for(int n1=1,a,b,c;n1<n;n1++)
    {
        cin>>a>>b>>c;
        add(a,b,c);
        add(b,a,c);
    }
    ME0(dp);
    DP(1,0,q);
    cout<<dp[1][q]<<endl;
    return 0;
}

P2014 选课

AC代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<string>
#include<queue>
#include<vector>
#include<cmath>
#include<stack>
#include<map>
#include<set>
using namespace std;
#define LL long long
const int MOD=100000007;
const int inf=0x3f3f3f3f;
const LL inff=0x3f3f3f3f3f3f3f3f;
const LL N=300005;
const LL M=5005;
#define MEF(x) memset(x,-1,sizeof(x))
#define ME0(x) memset(x,0,sizeof(x))
#define MEI(x) memset(x,inf,sizeof(x))
struct Edge
{
    int u,v,w,next;
}edge[605];
int cnt,first[305];
void init()
{
    cnt=0;
    MEF(first);
}
void add(int u,int v,int w)
{
    edge[cnt].u=u;
    edge[cnt].v=v;
    edge[cnt].w=w;
    edge[cnt].next=first[u];
    first[u]=cnt++;
}
int dp[305][305];
void Dp(int x,int m)
{
    for(int i=first[x];i!=-1;i=edge[i].next)
    {
        int lv=edge[i].v;
        dp[lv][1]=edge[i].w;
        Dp(lv,m);
        for(int j=m;j>=1;j--)
        {
            for(int k=0;k<=j;k++)
            {
                if((j!=1&&j!=k)||x==0)
                {
                    dp[x][j]=max(dp[x][j],dp[x][j-k]+dp[lv][k]);
                }
            }
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);
    int n,m;
    cin>>n>>m;
    init();
    for(int n1=1,a,b;n1<=n;n1++)
    {
        cin>>a>>b;
        add(a,n1,b);
    }
    Dp(0,m);
    cout<<dp[0][m]<<endl;
    return 0;
}