LeetCode每日一题 1222. 可以攻击国王的皇后

题目描述

在一个 8x8 的棋盘上，放置着若干「黑皇后」和一个「白国王」。给定一个由整数坐标组成的数组 queens ，表示黑皇后的位置；以及一对坐标 king ，表示白国王的位置，需要返回所有可以攻击国王的皇后的坐标(任意顺序)。

解题思路

我们可以模拟国王在棋盘上的移动，并检查每个方向上是否有皇后可以攻击国王。我们需要定义八个可能的移动方向，包括向上、向下、向左、向右以及四个对角线方向。对于每个方向，我们从国王的位置开始，按照该方向进行移动，同时检查是否有皇后。如果找到皇后，我们将其坐标添加到结果中。

代码实现

class Solution {
public:
    vector<vector<int>> queensAttacktheKing(vector<vector<int>>& queens, vector<int>& king) {
        vector<vector<int>> result;
        vector<vector<bool>> board(8, vector<bool>(8, false));

        // 标记皇后的位置
        for (const auto& queen : queens) {
            board[queen[0]][queen[1]] = true;
        }

        int kingX = king[0];
        int kingY = king[1];

        // 八个可能的移动方向
        vector<vector<int>> directions = {
            {-1, 0}, {1, 0}, {0, -1}, {0, 1},
            {-1, -1}, {-1, 1}, {1, -1}, {1, 1}
        };

        // 检查每个方向
        for (const auto& dir : directions) {
            int dx = dir[0];
            int dy = dir[1];
            int x = kingX + dx;
            int y = kingY + dy;

            // 沿着当前方向移动，直到越界或找到皇后
            while (x >= 0 && x < 8 && y >= 0 && y < 8) {
                if (board[x][y]) {
                    result.push_back({x, y});
                    break;
                }
                x += dx;
                y += dy;
            }
        }

        return result;
    }
};