LeetCode 1222. 可以攻击国王的皇后

【LetMeFly】1222.可以攻击国王的皇后

力扣题目链接：https://leetcode.cn/problems/queens-that-can-attack-the-king/

在一个 8x8 的棋盘上，放置着若干「黑皇后」和一个「白国王」。

给定一个由整数坐标组成的数组 queens ，表示黑皇后的位置；以及一对坐标 king ，表示白国王的位置，返回所有可以攻击国王的皇后的坐标(任意顺序)。

示例 1：

输入：queens = [[0,1],[1,0],[4,0],[0,4],[3,3],[2,4]], king = [0,0]
输出：[[0,1],[1,0],[3,3]]
解释： 
[0,1] 的皇后可以攻击到国王，因为他们在同一行上。 
[1,0] 的皇后可以攻击到国王，因为他们在同一列上。 
[3,3] 的皇后可以攻击到国王，因为他们在同一条对角线上。 
[0,4] 的皇后无法攻击到国王，因为她被位于 [0,1] 的皇后挡住了。 
[4,0] 的皇后无法攻击到国王，因为她被位于 [1,0] 的皇后挡住了。 
[2,4] 的皇后无法攻击到国王，因为她和国王不在同一行/列/对角线上。

示例 2：

输入：queens = [[0,0],[1,1],[2,2],[3,4],[3,5],[4,4],[4,5]], king = [3,3]
输出：[[2,2],[3,4],[4,4]]

示例 3：

输入：queens = [[5,6],[7,7],[2,1],[0,7],[1,6],[5,1],[3,7],[0,3],[4,0],[1,2],[6,3],[5,0],[0,4],[2,2],[1,1],[6,4],[5,4],[0,0],[2,6],[4,5],[5,2],[1,4],[7,5],[2,3],[0,5],[4,2],[1,0],[2,7],[0,1],[4,6],[6,1],[0,6],[4,3],[1,7]], king = [3,4]
输出：[[2,3],[1,4],[1,6],[3,7],[4,3],[5,4],[4,5]]

提示：

• 1 <= queens.length <= 63
• queens[i].length == 2
• 0 <= queens[i][j] < 8
• king.length == 2
• 0 <= king[0], king[1] < 8
• 一个棋盘格上最多只能放置一枚棋子。

方法一：哈希 + 模拟

皇后可以八个方向移动（↑↓←→↖↙↗↘），不能跳过其他棋子。

首先将所有的“皇后”的位置放入哈希表中，以便$O(1)$的时间复杂度查询某个位置是否有皇后。

接着从国王位置开始往8个方向遍历，遍历到皇后或超出棋盘边界为止。若遇到皇后，则加入到答案中。

• 时间复杂度$O(C^2)$，其中$C=8$
• 空间复杂度$O(len(queen))$

AC代码

C++

class Solution {
    
    
public:
    vector<vector<int>> queensAttacktheKing(vector<vector<int>>& queens, vector<int>& king) {
    
    
        unordered_set<int> se;
        for (auto& q : queens) {
    
    
            se.insert(q[0] * 100 + q[1]);
        }

        vector<vector<int>> ans;
        for (int dx = -1; dx <= 1; dx++) {
    
    
            for (int dy = -1; dy <= 1; dy++) {
    
    
                if (!dx && !dy) {
    
    
                    continue;
                }
                int nowX = king[0], nowY = king[1];
                while (nowX + dx >= 0 && nowX + dx < 8 && nowY + dy >= 0 && nowY + dy < 8) {
    
    
                    nowX += dx, nowY += dy;
                    if (se.count(nowX * 100 + nowY)) {
    
    
                        ans.push_back({
    
    nowX, nowY});
                        break;
                    }
                }
            }
        }
        return ans;
    }
};

Python

# from typing import List

class Solution:
    def queensAttacktheKing(self, queens: List[List[int]], king: List[int]) -> List[List[int]]:
        se = set(map(tuple, queens))
        ans = []
        for dx in range(-1, 2):
            for dy in range(-1, 2):
                if not dx and not dy:
                    continue
                nowX, nowY = king
                while 0 <= nowX + dx < 8 and 0 <= nowY + dy < 8:
                    nowX, nowY = nowX + dx, nowY + dy
                    if (nowX, nowY) in se:
                        ans.append([nowX, nowY])
                        break
        return ans

同步发文于CSDN，原创不易，转载经作者同意后请附上原文链接哦~
Tisfy：https://letmefly.blog.csdn.net/article/details/132870626