[CTSC1997] 选课
作者:互联网
一道典型的树型dp
首先每门课肯定只能选择一次,那么毫无疑问是一道树上的01背包,容量可以看作科目的门数
设计状态:\(f[i][j]代表了以i为根节点的子树中选择j门课程获得的最大得分\)
转移:对于节点\(x\),进行01背包的转移(记得倒序!记得倒序!记得倒序!)
小细节:对于每个节点的转移,若它不是0节点,要将这个节点的学分再加上
贴码
#include <bits/stdc++.h>
#define pb push_back
using namespace std;
const int N=310;
int n,m,s[N],f[N][N];//f[i][j]代表以i为根的子树中取j个节点取得的最大价值
vector<int> v[N];
void dp(int x)
{
f[x][0]=0;
for(int i=0;i<v[x].size();i++)
{
int to=v[x][i];
dp(to);
for(int t=m;t>=0;t--)//当前选课的总门数
for(int j=t;j>=0;j--)//子节点选课门数
if(t-j>=0)
f[x][t]=max(f[x][t],f[x][t-j]+f[to][j]);
}
if(x!=0)
for(int t=m;t>0;t--)
f[x][t]=f[x][t-1]+s[x];//若非0节点,就要加上它本身的学分
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
int x;
cin>>x>>s[i];
v[x].pb(i);
}
memset(f,0xcf,sizeof(f));
dp(0);
cout<<f[0][m]<<endl;
return 0;
}
标签:CTSC1997,选课,int,--,倒序,节点,dp 来源: https://www.cnblogs.com/sheepcsy/p/16420142.html