题意:
给出n*n的数,要求从1,1到n,n,且沿路不可有超过k个的负值,求最大的总和
思路:
用一个数组标记考虑的情况是否已经算出,一个数组标记是否位置重复,一开始觉得只要map初始化为inf然后如果不是inf的化就是经过计算了,不需要vis,后来发现就算经过计算map也会是inf,所以行不通,还是得要两个数组来标记
dp函数就是计算如果采用data[q][w]的话,那么最优解是什么.
值得注意的是最后一格是必须要用到的,直接返回,且一开始必须考虑到值得正负性
用到了回溯,因为 要考虑所有的可能
代码:
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#define LL long long
#define INF -200000000
using namespace std;
const int maxn = 80;
int data[maxn][maxn];
int dir[3][2] = {{0, -1}, {1, 0}, {0, 1}};
int gird[maxn][maxn];
long long map[maxn][maxn][7][3];
int vis[maxn][maxn][7][3];
int n, k;
bool check(int x, int y) {
if(x >=1 && x <=n && y>=1 && y<=n) return true;
else return false;
}
long long dp(int q, int w, int r, int d) {
int &flag = vis[q][w][r][d];//表示这种情况是否被算出
long long &res = map[q][w][r][d];//如果最优解已经算出来了,那么直接返回
if(flag) return res;
if(r == 0) {
flag = 1;
res = INF;
return res;
}
if(q==n && w==n) {//最后一格直接返回数值(必须采纳)
flag = 1;
res = data[q][w];
return res;
}
res = INF;
for(int i=0; i<3; i++) {
if(d==2 && i==0 || d==0 && i==2) continue;
int x = q + dir[i][0];
int y = w + dir[i][1];
int t = 0;
if(data[q][w] < 0) t = -1;
if(check(x, y) && !gird[x][y]) {//能够走的格子
gird[x][y] = 1;//标记被访问的点
long long temp = dp(x, y, r+t, i);//之后的所有路线
if(temp != INF)
res = max(res, temp+data[q][w]);采纳当前的格子的最优解
gird[x][y] = 0;//回溯
}
}
flag = 1;//情况已经计算过了
return res;
}
int main() {
int kase = 1;
while(scanf("%d%d", &n, &k) != EOF && n+k) {
for(int i=1; i<=n; i++)
for(int j=1; j<=n; j++) {
scanf("%d", &data[i][j]);
}
memset(vis, 0, sizeof(vis));
memset(gird, 0, sizeof(gird));
if(data[n][n] < 0) k--;//最后一个是直接返回的,没有考虑到k值正负,所以先考虑
long long res = dp(1, 1, k+1, 1);//1是避免与2,0冲突
printf("Case %d: ", kase++);
if(res != INF) printf("%lld\n", res);
else printf("impossible\n");
}
return 0;
}