【
在一个 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] 的皇后无法攻击到国王,因为她和国王不在同一行/列/对角线上。
提示:
1 <= queens.length <= 63
queens[i].length == 2
0 <= queens[i][j] < 8
king.length == 2
0 <= king[0], king[1] < 8
一个棋盘格上最多只能放置一枚棋子
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/queens-that-can-attack-the-king
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
】
矩阵题,首先想到方向数组和DFS,不过这道题不能用DFS。
这道题因为就一个国王,多个皇后,从皇后角度思考怎么攻击国王不怎么好处理,那么从国王角度思考怎么能被皇后攻击到,思路就明确了。但不能用DFS,用while(1)就行,8条射线,能到达皇后的地方,必然能被皇后攻击到。
int xp[8] = {0, 0, 1, 1, 1, -1, -1, -1};
int yp[8] = {1, -1, 0, 1, -1, 0, 1, -1};
int** queensAttacktheKing(int** queens, int queensSize, int* queensColSize, int* king, int kingSize, int* returnSize, int** returnColumnSizes){
int **retarr;
int i;
int idx = 0;
int allarr[8][8] = {0};
int x = king[0];
int y = king[1];
int flag;
retarr = (int **)malloc(sizeof(int *) * 8); // 根据题意,皇后小于8个
for (i = 0; i < 8; i++) {
retarr[i] = (int *)malloc(sizeof(int) * 2);
}
for (i = 0; i < queensSize; i++) {
allarr[queens[i][0]][queens[i][1]] = 1;
}
for (i = 0; i < 8; i++) {
flag = 0;
x = king[0]; // 国王的位置
y = king[1];
while (1) { // 开始以国王的位置,8条射线辐射,能到达皇后的位置,即说明能被皇后攻击
x += xp[i];
y += yp[i];
// 这里之前写错,x>8 导致allarr[x][y]栈溢出
if (x < 0 || x >= 8 || y < 0|| y >= 8) {
break;
}
if (allarr[x][y] == 1) {
flag = 1;
break;
}
}
if (flag == 1) {
retarr[idx][0] = x;
retarr[idx][1] = y;
idx++;
}
}
*returnColumnSizes = queensColSize;
*returnSize = idx;
return retarr;
}