LeetCode刷题日记（2596. 检查骑士巡视方案）

'''
骑士在一张 n x n 的棋盘上巡视。在有效的巡视方案中
骑士会从棋盘的左上角出发，并且访问棋盘上的每个格子 恰好一次 。
给你一个 n x n 的整数矩阵 grid ，由范围 [0, n * n - 1] 内的不同整数组成，
其中 grid[row][col] 表示单元格 (row, col) 是骑士访问的第 grid[row][col] 个单元格。
骑士的行动是从下标 0 开始的。
如果 grid 表示了骑士的有效巡视方案，返回 true；否则返回 false。
注意，骑士行动时可以垂直移动两个格子且水平移动一个格子，
或水平移动两个格子且垂直移动一个格子。下图展示了骑士从某个格子出发可能的八种行动路线。
'''
from typing import List


class Solution:
    # 暴力模拟
    def checkValidGrid(self, grid: List[List[int]]) -> bool:
        n = len(grid)
        step = n * n - 1
        row = col = 0
        if grid[row][col] != 0:
            return False

        for i in range(1, step + 1):
            if row - 2 >= 0 and col - 1 >= 0 and grid[row - 2][col - 1] == i:
                row -= 2
                col -= 1
            elif row - 2 >= 0 and col + 1 < n and grid[row - 2][col + 1] == i:
                row -= 2
                col += 1
            elif row - 1 >= 0 and col - 2 >= 0 and grid[row - 1][col - 2] == i:
                row -= 1
                col -= 2
            elif row - 1 >= 0 and col + 2 < n and grid[row - 1][col + 2] == i:
                row -= 1
                col += 2
            elif row + 1 < n and col - 2 >= 0 and grid[row + 1][col - 2] == i:
                row += 1
                col -= 2
            elif row + 1 < n and col + 2 < n and grid[row + 1][col + 2] == i:
                row += 1
                col += 2
            elif row + 2 < n and col - 1 >= 0 and grid[row + 2][col - 1] == i:
                row += 2
                col -= 1
            elif row + 2 < n and col + 1 < n and grid[row + 2][col + 1] == i:
                row += 2
                col += 1
            else:
                return False

        return True
    # 优化：将每个数字的位置存储起来，然后判断相邻两个数字的位置差是否为(1,2)或(2,1)
    def checkValidGrid2(self, grid: List[List[int]]) -> bool:
        if grid[0][0] != 0:
            # 左上角不为0
            return False

        n = len(grid)
        # 用来存储每个数字的位置
        pos = [(0, 0) for _ in range(n * n)]

        for i in range(n):
            for j in range(n):
                # pos[k] = (x,y) 表示数字k在位置(x,y)
                pos[grid[i][j]] = (i, j)

        for i in range(len(pos) - 1):
            x1, y1 = pos[i]
            x2, y2 = pos[i + 1]
            # 相邻两个数字的位置差不为(1,2)或(2,1)
            if not (abs(x2 - x1) == 1 and abs(y2 - y1) == 2 or abs(x2 - x1) == 2 and abs(y2 - y1) == 1):
                return False
        return True