問題 B: 【動態規劃】聖誕樹
題目描述
聖誕特別禮物掛在一棵聖誕樹上,這棵樹有n層,每層有一件禮物,每件禮物都有一個價值,有的禮物還有一些連接線,與下層的禮物相連。領取禮物的規則如下:任選一件禮物,它的下面如果有連接線,則可以繼續取它連接的禮物,依此類推直至取到沒有連接線的禮物才結束。你如果是第一個去取,怎樣取才能獲得最大的價值呢?請你編一程序解決這一問題。
輸入
第1行只有一個數據n(n≤100),表示有n層禮物,以下有n行數據,分別表示第1~n層禮物的狀態,每行至少由一個數據構成,且第一個數據表示該禮物的價值,後面的數據表示它與哪些層的禮物相連,如果每行只有一個數據則說明這層禮物沒有與下層禮物相連,每個數據大小均不超過10000。
輸出
只有一個數,表示獲得的最大價值。
樣例輸入 Copy
3
12 2 3
20
30
樣例輸出 Copy
42
先上代碼。
#include <bits/stdc++.h>
#pragma GCC optimize(2)
using namespace std;
typedef long long ll;
int a[105][105],b[105],maxn,dp[105];
int main(){
int n,k,temp;
char t;
cin>>n;
for(int i=0;i<n;i++){
k=0;
cin>>b[i];
scanf("%c",&t);
while(t==' '){
cin>>a[i][k++];
scanf("%c",&t);
}
}
//調試
// cout<<a[0][0]<<' '<<a[0][1]<<' '<<a[1][0]<<endl;
// cout<<a[2][0]<<' '<<a[0][2]<<' '<<a[3][0]<<endl;
// cout<<b[0]<<' '<<b[1]<<' '<<b[2]<<endl;
dp[n-1]=b[n-1];
maxn=dp[n-1];
for(int i=n-2;i>=0;i--){
dp[i]=b[i];
for(int j=0;;j++){
if(a[i][j]==0) break;
temp=b[i]+dp[a[i][j]-1];
dp[i]=max(dp[i],temp);
}
maxn=max(dp[i],maxn);
}
cout<<maxn<<endl;
}
本題的不定輸出有點煩,但我以前一直栽在緩衝區的空白,利用這一特點,便可解決。
ps:輸入一個整形數(不一定是整形,其他也可以)後,遇到空白(空格、tab、換行符)停止,但這些空白是留在緩衝區的,一般的話,如果下一個輸入的數是整形什麼的話,會自動跳過,唯獨要是字符類型就糟了,這空白會賦值給這字符。但反過來可以利用這點,達到不定輸入。
再補充一下cin會自動跳過空白,用樸素的scanf("%c",&t);或t=getchar();纔可以。所以也可以這麼寫。
cin>>b[i];
t=getchar();
while(t==' '){
cin>>a[i][k++];
t=getchar();
}
同時,輸入整形數時,前面有無數個空格都沒事,所以輸入時“33 (中間不止一個空格) 44”。也是同樣的效果,這應該就是個基礎知識。
當然還有更簡化的。
int n=0;
while(scanf("%d",&a[++n])!=EOF)
;
n--;
for(int i=1;i<=n;i++) cout<<a[i]<<' ';
如果一道題中只有一行不定輸入,可以用這個,在oj上也可以過,但這題不行,它有多組不定輸入,每次不定輸入需要加ctrl+Z才結束(換行符結束不了)
這是上面代碼的實現:
就dp而言,這題看出是逆向dp,狀態轉移公式很容易找出。水題一道。