九度oj 题目1335：闯迷宫

题目描述：

sun所在学校每年都要举行电脑节，今年电脑节有一个新的趣味比赛项目叫做闯迷宫。
sun的室友在帮电脑节设计迷宫，所以室友就请sun帮忙计算下走出迷宫的最少步数。
知道了最少步数就可以辅助控制比赛难度以及去掉一些没有路径到达终点的map。
比赛规则是：从原点（0,0）开始走到终点（n-1,n-1），只能上下左右4个方向走，只能在给定的矩阵里走。

输入：

输入有多组数据。
每组数据输入n（0<n<=100），然后输入n*n的01矩阵，0代表该格子没有障碍，为1表示有障碍物。
注意：如果输入中的原点和终点为1则这个迷宫是不可达的。

输出：

对每组输入输出该迷宫的最短步数，若不能到达则输出-1。

样例输入：

2
0 1
0 0
5
0 0 0 0 0
1 0 1 0 1
0 0 0 0 0
0 1 1 1 0
1 0 1 0 0

样例输出：

2
8

开始用dfs做的，结果超时了
代码如下：

#include <cstdio>
#include <cstdlib>
#include <string>
#include <cstring>
#include <algorithm>
#include <cmath>
#define MAX 102
#define inf 100002

int dir[][2] = {{0,1},{0,-1},{-1,0},{1,0}};
int map[MAX][MAX];
int flag[MAX][MAX];
int n;
int min;

void dfs(int x, int y, int cnt) {
    if(x == n-1 && y == n-1) {
        if(min > cnt) {
            min = cnt;
        }
        return;
    }
    if(cnt > min) {
        return;
    }
    for(int i = 0; i < 4; i++) {
        int tempx = x + dir[i][0];
        int tempy = y + dir[i][1];
        if(tempx >= 0 && tempx < n && tempy >= 0 && tempy < n && flag[tempx][tempy] == 0 &&map[tempx][tempy] == 0) {
            flag[tempx][tempy] = 1;
            dfs(tempx, tempy, cnt+1); 
            flag[tempx][tempy] = 0;
        }
    }
}

int main(int argc, char const *argv[])
{
    freopen("input.txt","r",stdin);
    while(scanf("%d",&n) != EOF) {
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < n; j++) {
                scanf("%d",&map[i][j]);
            }
        }
        if(map[0][0] == 1|| map[n-1][n-1] == 1) {
            puts("-1");
            continue;
        }
        min = inf;
        memset(flag,0,sizeof(flag));
        dfs(0, 0, 0);
        if(min != inf) {
            printf("%d
", min);
        }
        else {
            puts("-1");
        }
        
    }
    return 0;
}

后来改成bfs,代码如下

#include <cstdio>
#include <cstdlib>
#include <string>
#include <cstring>
#include <algorithm>
#include <queue>
#define MAX 102
#define inf 100002

using namespace std;

int dir[][2] = {{0,1},{0,-1},{-1,0},{1,0}};
int map[MAX][MAX];
int cnt[MAX][MAX];
int n;
int ans;

struct Point{
    int x;
    int y;
};

queue <Point> temp;

void bfs() {
    Point source;
    source.x = 0, source.y = 0;
    cnt[0][0] = 0;
    temp.push(source);
    while(!temp.empty()) {
        Point tmp = temp.front();
        temp.pop();
        for(int i = 0; i < 4; i++) {
            int tempx = tmp.x + dir[i][0];
            int tempy = tmp.y + dir[i][1];
            if(tempx >= 0 && tempx < n && tempy >= 0 && tempy < n && map[tempx][tempy] == 0) {
                int tcnt = cnt[tmp.x][tmp.y] + 1;
                if(tcnt < cnt[tempx][tempy]) {
                    Point p;
                    p.x = tempx;
                    p.y = tempy;
                    cnt[tempx][tempy] = tcnt;
                    temp.push(p);
                }
            }
        }
    }
}
    

int main(int argc, char const *argv[])
{
    //freopen("input.txt","r",stdin);
    while(scanf("%d",&n) != EOF) {
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < n; j++) {
                scanf("%d",&map[i][j]);
                cnt[i][j] = inf;
            }
        }
        if(map[0][0] == 1|| map[n-1][n-1] == 1) {
            puts("-1");
            continue;
        }
        bfs();
        ans = cnt[n-1][n-1];
        if(ans != inf) {
            printf("%d
", ans);
        }
        else {
            puts("-1");
        }
        
    }
    return 0;
}

竟然一次通过，不错不错