樹形DP 洛谷P2014 選課
阿新 • • 發佈:2017-05-24
ret mem 獲得 學習課程 clu 兩個 tor 輸入輸出格式 scan
輸出格式:
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門課程的最大得分。
輸入輸出樣例
輸入樣例#1:7 4 2 2 0 1 0 4 2 1 7 1 7 6 2 2輸出樣例#1:
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 選課