luogu P2014 选课
42行代码或成此题最X题解?
一看:树形dp
f[i][j]表示以i为根的子树选了j个课所获得的最大学分
等等,课可以重复选!
要用背包搞搞喽
每次枚举子节点
对每个子节点先进行树形dp
再对父节点进行背包
这里可以把0看成一个假根,可以不选
所以对于每个不是假根的节点,一定要选自己
怎么选?
先背包,背包好了后f[x][i]=f[x][i-1]+s[i]
完美~
std:
#include<bits/stdc++.h>
using namespace std;
const int N=505;
int n,m;
struct node{//定义node
vector<int> q;
int val;
}s[N];
int f[N][N];//以i为根的子树选了j个课所获得的最大学分
void dfs(int x){
f[x][0]=0;//一个都不选
for(int i=0;i<s[x].q.size();i++){//遍历儿子
int y=s[x].q[i];
dfs(y);//先对儿子进行dfs
for(int j=m;j>=0;j--){//枚举x的空间
for(int k=j;k>=0;k--){//枚举y的空间
if(j-k>=0)
f[x][j]=max(f[x][j],f[x][j-k]+f[y][k]);
}
}
}
if(x!=0){//是自己要选的 (非假根)
for(int i=m;i>0;i--)
f[x][i]=f[x][i-1]+s[x].val;//选自己
}
for(int i=0;i<=m;i++) {
printf("%d %d %d\n",x,i,f[x][i]);
}
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
int x,y;
scanf("%d%d",&x,&y);
s[x].q.push_back(i);
s[i].val=y;
}
memset(f,0xcf,sizeof(f));//将初始值设为负的极大值
dfs(0);//从假根开始dfs
printf("%d\n",f[0][m]);//答案在假根里
return 0;
}