蓝桥杯 地宫取宝 记忆化搜索

  历届试题 地宫取宝  

时间限制：1.0s   内存限制：256.0MB

    

问题描述

　　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

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
#include <queue>

using namespace std;
#define maxn 55
#define inf 1000000007

typedef long long ll;
pair <int ,int > p;
int arr[maxn][maxn];
ll vis[maxn][maxn][15][15];
int n,m,k;
ll ans;
ll dfs(int cnt,int x, int y,int mo);
bool check(int x,int y);
int main()
{
	memset(vis,-1,sizeof(vis));
	scanf("%d%d%d",&n,&m,&k);
	for(int i = 0; i < n; i++){
		for(int j = 0; j < m; j++){
			scanf("%d",&arr[i][j]);
		}
	}
	dfs(0,0,0,-1);
	printf("%lld\n",vis[0][0][0][0]);
    return 0;
}
ll dfs(int cnt,int x,int y,int mo){
	if(vis[x][y][cnt][mo+1] != -1){
		return vis[x][y][cnt][mo+1];
	} 
	if(x == n-1 && y == m-1){
		if(cnt == k){
			vis[x][y][cnt][mo+1] = 1;
		}else if(cnt == k - 1 && arr[x][y] > mo){
			vis[x][y][cnt][mo+1] = 1;
		}else {
			vis[x][y][cnt][mo+1] = 0;
		}
		return vis[x][y][cnt][mo+1];
	}
	ll s = 0;
	if(arr[x][y] > mo){
		if(x+1 < n){
		    s = (s + dfs(cnt+1,x+1,y,arr[x][y])%inf +dfs(cnt,x+1,y,mo)%inf) % inf;
		}
		if(y+1 < m){
			s = (s + dfs(cnt+1,x,y+1,arr[x][y])%inf +dfs(cnt,x,y+1,mo)%inf) % inf;
		}
	}else{
		if(x+1 < n){
			s = (s + dfs(cnt,x+1,y,mo)%inf)%inf;
		}
		if(y+1 < m){
			s = (s+dfs(cnt,x,y+1,mo)%inf)%inf;
		}
	}
	
	vis[x][y][cnt][mo+1] = s;
	return vis[x][y][cnt][mo+1];
}

裸搜只能过三个点。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
#include <queue>

using namespace std;
#define maxn 55
#define inf 0x7ffffff
typedef long long ll;
pair <int ,int > p;
int arr[maxn][maxn];
int vis[maxn][maxn];
int dir[2][2] ={0,1,1,0};
int n,m,k,ans;
void dfs(int cnt,int x, int y,int mo);
bool check(int x,int y);
int main()
{
	memset(vis,0,sizeof(vis));
	scanf("%d%d%d",&n,&m,&k);
	for(int i = 0; i < n; i++){
		for(int j = 0; j < m; j++){
			scanf("%d",&arr[i][j]);
		}
	}
	ans = 0;
	vis[0][0] = 1;
	dfs(1,0,0,arr[0][0]);
	dfs(0,0,0,-1);
	printf("%d\n",ans);
    return 0;
}
void dfs(int cnt,int x,int y,int mo){
	if(x == n-1 && y == m-1){
		if(cnt == k){
			ans ++;
			if(ans > 1000000007){
				ans %= 1000000007;
			}
		}
		return ;
	}
	for(int i = 0; i < 2; i++){
		int tmpx = x + dir[i][0];
		int tmpy = y + dir[i][1];
		if(check(tmpx,tmpy)){
			vis[tmpx][tmpy] = 1;
			if(arr[tmpx][tmpy] > mo && cnt < k){
				dfs(cnt+1,tmpx,tmpy,arr[tmpx][tmpy]);
			}
			dfs(cnt,tmpx,tmpy,mo);
			vis[tmpx][tmpy] = 0;
		}
	}
}
bool check(int x,int y){
	if(x < 0 || x >= n || y < 0 || y >= m || vis[x][y] == 1){
		return false;
	}
	return true;
}