P2014 選課

題目描述

在大學裏每個學生，為了達到一定的學分，必須從很多課程裏選擇一些課程來學習，在課程裏有些課程必須在某些課程之前學習，如高等數學總是在其它課程之前學習。現在有N門功課，每門課有個學分，每門課有一門或沒有直接先修課（若課程a是課程b的先修課即只有學完了課程a，才能學習課程b）。一個學生要從這些課程裏選擇M門課程學習，問他能獲得的最大學分是多少？

輸入輸出格式

第一行有兩個整數N,M用空格隔開。(1<=N<=300,1<=M<=300)

接下來的N行,第I+1行包含兩個整數ki和si, ki表示第I門課的直接先修課，si表示第I門課的學分。若ki=0表示沒有直接先修課（1<=ki<=N, 1<=si<=20）。

只有一行，選M門課程的最大得分。

輸入輸出樣例

7  4
2  2
0  1
0  4
2  1
7  1
7  6
2  2

13

傷心且難過，vector寫不出來......

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 using namespace std;
 6 int n,m;
 7 int f[311][311];//以i為頂點 選取j門課程可得的最大學分 
 

 8 struct data{
 9     int l,r,ch;
10 }node[311];
11 
12 void insert(int i,int j,int k){//多叉樹轉二叉樹 左兒子右兄弟 
13      node[i].r=node[j].l;
14      node[j].l=i;
15      node[i].ch=k;
16      return;
17 }
18 
19 int dfs(int i,int j){//i為頂點 選j門課程 
20     if(!j||i<0) return 0;//都是i j惹的禍 
21     if(!i) return dfs(node[i].l,j);
 
22     if(f[i][j]>=0) return f[i][j];
23     f[i][j]=dfs(node[i].r,j);
24     for(int k=1;k<=j;k++) f[i][j]=max(f[i][j],dfs(node[i].l,k-1)+node[i].ch+dfs(node[i].r,j-k));  
25     return f[i][j];
26 }
27 
28 int main(){
29     scanf("%d%d",&n,&m);
30     int a=0,b=0;
31     memset(node,-1,sizeof(node));
32     memset(f,-1,sizeof(f));
33     for(int i=1;i<=n;i++){
34         scanf("%d%d",&a,&b);
35         insert(i,a,b);
36     }
37     printf("%d\n",dfs(0,m));
38     return 0;
39 }

樹形DP 洛谷P2014 選課