刷题使我快乐，满脸开心.jpg

• 来源：力扣（LeetCode）
• 链接：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
• 一个棋盘格上最多只能放置一枚棋子。

思路

从国王往八个方向向外找是否有皇后即可，且只能是第一个碰到的皇后可以攻击到国王
这里看到官方题解后有个小优化可以注意下：
可以先将皇后坐标转化为一个map，代码中的x * 8 + y，中这个8只要是大于8的数均可，这样可以避免直接比较坐标所带来的复杂度

自己的代码有点丑，就放相同思路的官方代码啦~

代码

func queensAttacktheKing(queens [][]int, king []int) [][]int {
    queen_pos := make(map[int]bool)
    for _, queen := range queens {
        x, y := queen[0], queen[1]
        queen_pos[x * 8 + y] = true
    }
        
    ans := [][]int{}
    for dx := -1; dx <= 1; dx++ {
        for dy := -1; dy <= 1; dy++ {
            if dx == 0 && dy == 0 {
                continue
            }
            kx, ky := king[0] + dx, king[1] + dy
            for kx >= 0 && kx < 8 && ky >= 0 && ky < 8 {
                pos := kx * 8 + ky
                if _, ok := queen_pos[pos]; ok {
                    ans = append(ans, []int{kx, ky})
                    break
                }
                kx += dx
                ky += dy
            }
        }
    }
    return ans
}