1. 程式人生 > >藍橋杯 地宮尋寶(DP or DP+記憶化搜索)

藍橋杯 地宮尋寶(DP or DP+記憶化搜索)

搜索 1.7 寫法 str 大眾 include import 行數據 哪些

藍橋杯 地宮尋寶

X 國王有一個地宮寶庫。是 n x m 個格子的矩陣。每個格子放一件寶貝。每個寶貝貼著價值標簽。

地宮的入口在左上角,出口在右下角。

小明被帶到地宮的入口,國王要求他只能向右或向下行走。

走過某個格子時,如果那個格子中的寶貝價值比小明手中任意寶貝價值都大,小明就可以拿起它(當然,也可以不拿)。

當小明走到出口時,如果他手中的寶貝恰好是k件,則這些寶貝就可以送給小明。

請你幫小明算一算,在給定的局面下,他有多少種不同的行動方案能獲得這k件寶貝。

【數據格式】

輸入一行3個整數,用空格分開:n m k (1<=n,m<=50, 1<=k<=12)

接下來有 n 行數據,每行有 m 個整數 Ci (0<=Ci<=12)代表這個格子上的寶物的價值

要求輸出一個整數,表示正好取k個寶貝的行動方案數。該數字可能很大,輸出它對 1000000007 取模的結果。

例如,輸入:
2 2 2
1 2
2 1
程序應該輸出:
2

再例如,輸入:
2 3 2
1 2 3
2 1 5
程序應該輸出:
14

資源約定:
峰值內存消耗(含虛擬機) < 256M
CPU消耗 < 2000ms


請嚴格按要求輸出,不要畫蛇添足地打印類似:“請您輸入...” 的多余內容。

所有代碼放在同一個源文件中,調試通過後,拷貝提交該源碼。
註意:不要使用package語句。不要使用jdk1.7及以上版本的特性。
註意:主類的名字必須是:Main,否則按無效代碼處理。

題解:

  先%一發我偉大的隊長 傑哥tql

  引入:m×n的方格,從左下角走到右上角,總共的路徑為C(m+n) m = C(m+n) n

 1 #include<cstdio>
 2 #define ll long long
 3 int val[58][58];
 4 ll C[118][118],dp[58][58][18]={0},mod=1000000007;
 5 //dp[x][y][k]表示x,y位置上的寶物作為第k個被拿的方案數 
 6 void init()//求組合數 
 7 {
 8     C[0][0]=1;
 9     for(int i=1;i<=100;i++)
10     {
11         C[i][0]=1
; 12 for(int j=1;j<=i;j++) 13 { 14 if(j<=i/2) 15 C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod; 16 else 17 C[i][j]=C[i][i-j]; 18 } 19 } 20 } 21 int main() 22 { 23 init(); 24 int n,m,K; 25 scanf("%d%d%d",&n,&m,&K); 26 for(int i=0;i<n;i++) 27 for(int j=0;j<m;j++) 28 scanf("%d",&val[i][j]); 29 ll ans=0; 30 for(int i=0;i<n;i++) 31 for(int j=0;j<m;j++) 32 { 33 //i,j位置上的寶物是第一個被拿的方案數即從開始地點走到i,j位置的方案數 34 dp[i][j][1]=C[i+j][i]; 35 for(int k=i;k>=0;k--)//這兩層循環表示i,j位置可以由哪些位置走到 36 for(int l=j;l>=0;l--) 37 { 38 if(val[k][l]<val[i][j]) 39 for(int t=2;t<=K;t++) 40 dp[i][j][t]=(dp[i][j][t]+(dp[k][l][t-1]*C[i-k+j-l][i-k])%mod)%mod;//C[i-k+j-l][i-k]相當於從k,l位置走到i,j位置的方案數 41 } 42 ans=(ans+(dp[i][j][K]*C[n-1-i+m-1-j][n-1-i])%mod)%mod;//從i,j位置到n-1,m-1位置的方案數 43 } 44 printf("%lld\n",ans); 45 return 0; 46 }

以及大眾化的記憶化搜索寫法:

 1 import java.util.Scanner;
 2 
 3 public class Main {
 4     static int[][][][] dp = new int[55][55][15][15];
 5     static int[][] a = new int[55][55];
 6     static int[][] vis = new int[55][55];
 7     static int[][] dir =  {{0,1},{1,0}};
 8     static int mod = 1000000007;
 9     static int n,m,k; 
10     static int dfs(int x,int y,int step,int maxv) {
11         if(dp[x][y][step][maxv]!=-1) 
12             return dp[x][y][ step][maxv];
13         int cnt = 0;
14         if(x==n-1&&y==m-1) { 
15             if((step == k-1&&maxv<a[x][y])||(step==k)) {
16                 return dp[x][y][step][maxv] = 1;
17             }
18             return dp[x][y][step][maxv] = 0;
19         }
20         for(int i=0;i<2;i++) {
21             int tx = x+dir[i][0];
22             int ty = y+dir[i][1];
23             if(tx<0||ty<0||tx>=n||ty>=m||vis[tx][ty]==1) {
24                 continue;
25             }
26             vis[tx][ty] = 1;
27             if(a[x][y]>maxv) {
28                 cnt = (cnt+dfs(tx, ty, step+1, a[x][y]))%mod;
29             }
30             cnt = (cnt + dfs(tx, ty, step, maxv))%mod;
31             vis[tx][ty] = 0;
32         }
33         return dp[x][y][step][maxv] = cnt%mod;
34     }
35     public static void main(String[] args) {
36         Scanner cin = new Scanner(System.in);
37         n = cin.nextInt();
38         m = cin.nextInt();
39         k = cin.nextInt();
40         for(int i=0;i<55;i++)
41             for(int j=0;j<55;j++)
42                 for(int k=0;k<15;k++)
43                     for(int l=0;l<15;l++)
44                         dp[i][j][k][l] = -1;
45         for(int i=0;i<n;i++)
46             for(int j=0;j<m;j++) {
47                 a[i][j] = cin.nextInt();
48                 a[i][j]++;
49             }
50                 
51         
52         System.out.println(dfs(0,0,0,0));
53     }
54 
55 }

藍橋杯 地宮尋寶(DP or DP+記憶化搜索)