hdu 1045 || zoj 1002 Fire Net(搜索:DFS+水题)

很水的一道题,直接DFS即可

我的思路是假设当前位置可以放子弹

不放子弹搜一遍,放上子弹搜一遍并更新行列状态

但是有一个问题就是在当前位置放置子弹后,如何在深搜的过程中还原为不放子弹的状态

我开始的想法是用一个bool vis[]来表示当前位置是否可以放子弹

但这样是不行的,因为在搜索的过程中这个数组被多次赋值了,不可能再回到原来的状态了

之后我的想法是用一个三维数组cur[pos][x][y]来保存遍历到pos这个位置时图的状态

这样就很好做了,搜完放子弹状态后在把cur数组赋值给原来的数组,回到原来状态即可

代码如下:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define MAXN 5
using namespace std;

int n, ans = -1;
int g[MAXN][MAXN], cur[17][MAXN][MAXN];

void change(int pos) {
int x = pos/n;
int y = pos%n;
g[x][y] = -1;
int tmp = x;
while(tmp>=1 && g[--tmp][y]==0) {
g[tmp][y] = -1;
}
tmp = x;
while(tmp<n-1 && g[++tmp][y]==0) {
g[tmp][y] = -1;
}
tmp = y;
while(tmp>=1 && g[x][--tmp]==0) {
g[x][tmp] = -1;
}
tmp = y;
while(tmp<n-1 && g[x][++tmp]==0) {
g[x][tmp] = -1;
}
}

void unchange(int pos) {
for(int i=0; i<n; ++i)
for(int j=0; j<n; ++j)
g[i][j] = cur[pos][i][j];
}

void dfs(int pos, int times) {
if(pos == n*n) {
ans = max(ans, times);
return ;
}

for(int i=0; i<n; ++i)
for(int j=0; j<n; ++j)
cur[pos][i][j] = g[i][j];

dfs(pos+1, times);
if(g[pos/n][pos%n] == 0) {
change(pos);
dfs(pos+1, times+1);
unchange(pos);
}
return ;
}

int main(void) {
char ch;
while(scanf("%d", &n) && n) {
ans = -1;
memset(g, 0, sizeof(g));
memset(cur, 0, sizeof(cur));
for(int i=0; i<n; ++i) {
getchar();
for(int j=0; j<n; ++j) {
ch = getchar();
if(ch == 'X')
g[i][j] = -1;
}
}
dfs(0, 0);
cout << ans << endl;
}
return 0;
}