Medium

The Knight’s TourC++

Full explanation · Time O(m * n) · Space O(1)

// Time:  O(m * n)
// Space: O(1)

// backtracking, greedy, warnsdorff's rule
class Solution {
public:
    vector<vector<int>> tourOfKnight(int m, int n, int r, int c) {
        static const vector<pair<int, int>> DIRECTIONS = {{1, 2}, {-1, 2}, {1, -2}, {-1, -2},
                                                          {2, 1}, {-2, 1}, {2, -1}, {-2, -1}};

        vector<vector<int>> result(m, vector<int>(n, -1));
        result[r][c] = 0;
        const function<bool(int, int, int)> backtracking = [&](int r, int c, int i) {
            const auto& degree = [&](const auto& x) {
                int cnt = 0;
                const auto& [r, c] = x;
                for (const auto& [dr, dc] : DIRECTIONS) {
                    const int nr = r + dr, nc = c + dc;
                    if (0 <= nr && nr < m && 0 <= nc && nc < n && result[nr][nc] == -1) {
                        ++cnt;
                    }
                }
                return cnt;
            };

            if (i == m * n) {
                return true;
            }
            vector<pair<int, int>> candidates;
            for (const auto& [dr, dc] : DIRECTIONS) {
                const int nr = r + dr, nc = c + dc;
                if (0 <= nr && nr < m && 0 <= nc && nc < n && result[nr][nc] == -1) {
                    candidates.emplace_back(nr, nc);
                }
            }
            sort(begin(candidates), end(candidates), [&](const auto& a, const auto& b) {
                return degree(a) < degree(b);
            });
            for (const auto& [nr, nc] : candidates) {  // warnsdorff's rule
                result[nr][nc] = i;
                if (backtracking(nr, nc, i + 1)) {
                    return true;
                }
                result[nr][nc] = -1;
            }
            return false;
        };

        backtracking(r, c, 1);
        return result;
    }
};

// Time:  O(8^(m * n - 1))
// Space: O(m * n)
// backtracking
class Solution2 {
public:
    vector<vector<int>> tourOfKnight(int m, int n, int r, int c) {
        static const vector<pair<int, int>> DIRECTIONS = {{1, 2}, {-1, 2}, {1, -2}, {-1, -2},
                                                          {2, 1}, {-2, 1}, {2, -1}, {-2, -1}};

        vector<vector<int>> result(m, vector<int>(n, -1));
        result[r][c] = 0;
        const function<bool(int, int, int)> backtracking = [&](int r, int c, int i) {
            if (i == m * n) {
                return true;
            }
            for (const auto& [dr, dc] : DIRECTIONS) {
                const int nr = r + dr, nc = c + dc;
                if (!(0 <= nr && nr < m && 0 <= nc && nc < n && result[nr][nc] == -1)) {
                    continue;
                }
                result[nr][nc] = i;
                if (backtracking(nr, nc, i + 1)) {
                    return true;
                }
                result[nr][nc] = -1;
            }
            return false;
        };

        backtracking(r, c, 1);
        return result;
    }
};