参考博客: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;
}