华为机考真题 -- 跳马

题目描述：

马是象棋(包括中国象棋和国际象棋)中的棋子，走法是每步直一格再斜一格，即先横着或者直者走一格，然后再斜着走一个对角线，可进可退，可越过河界，俗称"马走日"字。给定 m 行 n 列的棋盘（网格图），棋盘上只有象棋中的棋子“马”，并且每个棋子有等级之分，等级为 k 的马可以跳 1~k 步(走的方式与象棋中“马”的规则一样，不可以超出棋盘位置)，问是否能将所有马跳到同一位置，如果存在，输出最少需要的总步数(每匹马的步数相加)，不存在则输出-1。

注：允许不同的马在跳的过程中跳到同一位置，坐标为(x,y)的马跳一次可以跳到的坐标为：(x+1, y+2)，(x+1, y-2)，(x+2, y+1)，(x+2, y-1)，(x-1, y+2)，(x-1, y-2)，(x-2, y+1)，(x-2, y-1)的格点上，但是不可以超出棋盘范围。

输入描述：

第一行输入m，n，代表 m 行 n 列的网格图棋盘（1 ≤ m, n ≤ 25）

接下来输入 m 行 n 列的网格图棋盘，如果第 i 行，第 j 列的元素为 "." ，代表此格点没有棋子，如果为数字 k(1 ≤ k ≤ 9)，代表此格点存在等级为 k 的“马”

输出描述：

输出最少需要的总步数(每匹马的步数相加)，不存在则输出-1。

示例1：

输入
3 2
..
2.
..

输出
0
说明：只有一匹马，不需要跳动

示例2：

输入
3 5
47.48
4744.
7....

输出
17

C++源码：

#include <iostream>
#include <vector>
#include <string>
#include <climits>

using namespace std;

// 函数声明
void dfs(int id, int x, int y, int val, int num);
void allStep(int n, int m, vector<string>& chessboard);
// 初始化dp数组，用于存储每匹马到达每个点的最小步数
vector<vector<vector<int>>> dp(500, vector<vector<int>>(20, vector<int>(20, INT_MAX)));

int n, m; // n, m用于存储棋盘的行数和列数

void allStep(int n, int m, vector<string>& chessboard) {

    int cnt = 0; // cnt用于计算棋盘上马的数量
    for (int x = 0; x < n; ++x) {
        for (int y = 0; y < m; ++y) {
            if (chessboard[x][y] == '.') { // 如果当前位置没有马，则跳过
                continue;
            }
            int k = chessboard[x][y] - '0'; // 读取并转换当前位置马的等级
            ++cnt; // 马的数量增加
            dfs(cnt, x, y, 0, k); // 从当前马的位置开始深度优先搜索，初始化移动次数为0
        }
    }
    int result = INT_MAX; // 初始化结果为一个很大的数
    for (int x = 0; x < n; ++x) {
        for (int y = 0; y < m; ++y) {
            int step = 0; // 用于计算所有马到达(x, y)的总步数
            for (int id = 1; id <= cnt; ++id) {
                step += dp[id][x][y]; // 累加每匹马到达(x, y)的步数
            }
            result = min(result, step); // 更新找到的最小步数
        }
    }
    if (result == INT_MAX) {
        cout << -1 << endl; // 如果ans仍然是初始值，则输出-1表示不可能
    }
    else {
        cout << result << endl;
    }
}

// x方向的移动增量，代表马的8种可能的移动
const vector<int> dx = { 1, 2, 2, 1, -1, -2, -2, -1 };
// y方向的移动增量，同上
const vector<int> dy = { 2, 1, -1, -2, -2, -1, 1, 2 };

void dfs(int id, int x, int y, int val, int num) {
    if (num < 0 || dp[id][x][y] <= val) { // 如果已经不可以移动，或当前记录的步数小于等于已找到的步数，则返回
        return;
    }
    dp[id][x][y] = val; // 更新到达当前位置的最小步数
    for (int i = 0; i < 8; ++i) { // 遍历马可以移动的8个方向
        int nx = x + dx[i];
        int ny = y + dy[i];
        if (nx < 0 || nx >= n || ny < 0 || ny >= m) { // 检查新位置是否越界
            continue;
        }
        dfs(id, nx, ny, val + 1, num - 1); // 递归调用dfs，移动次数增加，可移动的最大次数减少
    }
}

int main() {

    cin >> n >> m; // 从输入读取棋盘的大小
    vector<string> chessboard(n);
    for (int i = 0; i < n; ++i) {
        cin >> chessboard[i]; // 读取棋盘的每一行
    }
    allStep(n,m, chessboard);

    system("pause");
    return 0;
}