题目描述
在一个 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;
}
};