2018年北京信息科技大学第十届程序设计竞赛暨ACM选拔赛（A - PUBG）

链接： https://www.nowcoder.com/acm/contest/118/A
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld

题目描述

最近，喜爱ACM的PBY同学沉迷吃鸡，无法自拔，于是又来到了熟悉的ERANGEL。经过一番搜寻，PBY同学准备动身前往安全区，但是，地图中埋伏了许多LYB，PBY的枪法很差，希望你能够帮他找到一条路线，每次只能向上、下、左、右移动，尽可能遇到较少的敌人。

输入描述:

题目包含多组测试，请处理到文件结束；
第一行是一个整数n，代表地图的大小；
接下来的n行中，每行包含n个整数a，每个数字a代表当前位置敌人的数量；
1 < n <= 100，1 <= a <= 100，-1代表当前位置，-2代表安全区。

输出描述:

对于每组测试数据，请输出从当前位置到安全区所遇到最少的敌人数量，每个输出占一行。

题意很清楚，就是输出到达目的地的最短路径。初看此题，瞬间懵逼，平时我做的搜索题都是“非0即1”这种。这道题显然需要对以前的方法进行变化，我们可以这样，设一个二维数组，二维数组每一个点存到达这个点的（最短）路程。

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

using namespace std;

typedef pair<int, int> pii;

const int MAX_N = 200;
const int INF = 1e9;                                   //用来初始化d数组

int mp[MAX_N][MAX_N];
int d[MAX_N][MAX_N];                                   //存到达某个点的路程
int dir[4][2] = {{0, -1}, {-1, 0}, {0, 1}, {1, 0}};
int sx, sy;
int ex, ey;
int n;

int bfs()
{
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            d[i][j] = INF;
        }
    }
    queue<pii> que;
    que.push(pii(sx, sy));
    d[sx][sy] = 0;
    while (!que.empty()) {
        pii temp = que.front();
        que.pop();
        for (int i = 0; i < 4; ++i) {
            int nx = temp.first + dir[i][0];
            int ny = temp.second + dir[i][1];
            if ((nx >= 0 && nx < n) && (ny >= 0 && ny < n)) {
                if (d[nx][ny] > d[temp.first][temp.second] + mp[nx][ny]) {
                    d[nx][ny] = d[temp.first][temp.second] + mp[nx][ny];
                    que.push(pii(nx, ny));
                }
            }
        }
    }
    return d[ex][ey];
}

int main()
{
    while (~scanf("%d", &n)) {
        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < n; ++j) {
                scanf("%d", &mp[i][j]);
                if (mp[i][j] == -1) {
                    sx = i;
                    sy = j;
                    mp[i][j] = 0;
                }
                if (mp[i][j] == -2) {
                    ex = i;
                    ey = j;
                    mp[i][j] = 0;
                }
            }
        }
        printf("%d\n", bfs());
    }
    return 0;
}