(中石油練習)問題 B: 聖誕樹(dp,不定輸入,水題)

問題 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,狀態轉移公式很容易找出。水題一道。

發佈了18 篇原創文章 · 獲贊 16 · 訪問量 1339
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章