【動態規劃】練習題-挖地雷

原題鏈接

題目描述

在一個地圖上有NN個地窖(N \le 20)(N≤20),每個地窖中埋有一定數量的地雷。同時,給出地窖之間的連接路徑。當地窖及其連接的數據給出之後,某人可以從任一處開始挖地雷,然後可以沿着指出的連接往下挖(僅能選擇一條路徑),當無連接時挖地雷工作結束。設計一個挖地雷的方案,使某人能挖到最多的地雷。

輸入格式

有若干行。

第11行只有一個數字,表示地窖的個數NN。

第22行有NN個數,分別表示每個地窖中的地雷個數。

第33行至第N+1N+1行表示地窖之間的連接情況:

第33行有n-1n−1個數(00或11),表示第一個地窖至第22個、第33個、…、第nn個地窖有否路徑連接。如第33行爲1 1 0 0 0 … 011000…0,則表示第11個地窖至第22個地窖有路徑,至第33個地窖有路徑,至第44個地窖、第55個、…、第nn個地窖沒有路徑。

第44行有n-2n−2個數,表示第二個地窖至第33個、第44個、…、第nn個地窖有否路徑連接。

… …

第n+1n+1行有11個數,表示第n-1n−1個地窖至第nn個地窖有否路徑連接。(爲00表示沒有路徑,爲11表示有路徑)。

輸出格式

有兩行

第一行表示挖得最多地雷時的挖地雷的順序,各地窖序號間以一個空格分隔,不得有多餘的空格。

第二行只有一個數,表示能挖到的最多地雷數。

輸入輸出樣例

輸入
5
10 8 4 7 6
1 1 1 0
0 0 0
1 1
1
輸出
1 3 4 5
27

結題思路

本題從圖的角度來說應該是一個有向圖,(看輸入輸出樣例結果得出來的,懂的就這些了,還不會用圖的思想結題,以後會了再來補充)。從動態規劃角度來說,我的解題思路是這樣的,借鑑了之前解數字金字塔,通過倒推的方式來想: 用f[i]記錄從i出發能夠挖到的最多的地雷數,b[i][j]表示從i到j是否連通,a[i]表示i處的地雷數,那麼就有如下的狀態轉移方程:

if(b[i][j]{
   f[i]=max(f[i],f[j]+a[i])
}

有了狀態轉移方程,最值就能夠得出來了,題目中還要求輸出挖地雷最多時的順序,也就是需要記錄下每次的路徑,用pre[i]來存儲從i到達的下一點(有點像鏈表)。

完整代碼

#include<iostream>
#include <algorithm>
using namespace std;
int n;
int a[1000];
int f[1000],pre[1000];
bool b[1000][1000]; 
int main(){
	cin>>n;
	for(int i=1;i<=n;i++)
	    cin>>a[i];
	for(int i=1;i<n;i++)
	for(int j=i+1;j<=n;j++)
	{
		cin>>b[i][j];
	}
	f[n]=a[n];	
	for(int i=n-1;i>=1;i--)
	{
	    int fj=0,path=0;
		for(int j=i+1;j<=n;j++)
	    {
		    if(b[i][j]&& f[j]>fj){
			    fj=f[j];
			    path=j;
	  	   }	
	   }
	    f[i]=fj+a[i];
		pre[i]=path;
	}
	int k=1;
	for(int i=2;i<=n;i++)
	if(f[i]>f[k]) k=i;
	cout<<k<<" ";
	int m;
	m=f[k];
	k=pre[k];
	while(k){
		cout<<k<<" ";
		k=pre[k];
	}
	cout<<endl;
    cout<<m;
	return 0;
} 
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章