luogu P2014 选课

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;
} 

猜你喜欢

转载自blog.csdn.net/weixin_43872370/article/details/84779102
今日推荐