1222. 可以攻击国王的皇后
在一个 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
- 一个棋盘格上最多只能放置一枚棋子。
思路
确定国王的位置,向他的八个方向进行遍历
代码
class Solution {
public:
vector<vector<int>> queensAttacktheKing(vector<vector<int>>& queens, vector<int>& king) {
int m = king[0];
int n = king[1];
vector<vector<int>> arr; //将满足条件的皇后放在里面
vector<vector<int>> arr1(8, vector<int>(8, 0));//定义一个8*8的数组
for (auto& queen : queens) {
arr1[queen[0]][queen[1]] = 1; //将皇后的位置放在对应的坐标系中
}
for(int i=-1;i<=1;i++)
for(int j=-1;j<=1;j++) //将八个方向一次列出
{
if(i==0&&j==0) //当i=0 与j=0时,表示原点不进行计算
{
continue;
}
int dx=m+i;//x移动后的位置
int dy=n+j;//y移动后的位置
while (dx >= 0 && dx < 8 && dy >= 0 && dy < 8)
{
if (arr1[dx][dy] == 1)
{
arr.push_back({dx, dy});
break;
}
dx += i;
dy += j;
}
}
return arr;
}
};
注意:
原先没有这一步,导致一直超时。
if(i==0&&j==0) //当i=0 与j=0时,表示原点不进行计算
{
continue;
}