本文最后更新于 388 天前,其中的信息可能已经有所发展或是发生改变。
题目描述:
在一个 8×8 的棋盘上,放置着若干「黑皇后」和一个「白国王」。
给定一个由整数坐标组成的数组 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
- 一个棋盘格上最多只能放置一枚棋子。
解法一:模拟
算法思路:
我可以从国王的角度出发,依次枚举国王可能受到攻击的八个方向,让国王出发,国王在每个方向上遇到的第一个皇后就是可以攻击他的皇后。
记国王的位置为 (kx, ky),枚举的方向为 (dx, dy),那么我们不断地将 kx 加上 dx,将 ky 加上 dy,直到遇到皇后或者走出边界位置。
代码实现:
class Solution {
public List<List<Integer>> queensAttacktheKing(int[][] queens, int[] king) {
// 创建一个集合来存储皇后的位置
Set<Integer> queenPos = new HashSet<Integer>();
for(int[] queen : queens){
int x = queen[0], y = queen[1];
queenPos.add(x * 8 + y);
}
// 创建一个列表来存储可以攻击国王的皇后的位置
List<List<Integer>> ans = new ArrayList<List<Integer>>();
// 遍历所有可能的八个方向
for(int dx = -1; dx <= 1; dx++){
for(int dy = -1; dy <= 1; dy++){
if(dx == 0 && dy == 0){
continue;
}
// 计算国王行进的位置
int kx = king[0] + dx, ky = king[1] + dy;
// 检查位置是否在棋盘的边界之内
while(kx >= 0 && kx < 8 && ky >= 0 && ky < 8){
int pos = kx * 8 + ky;
// 检查该位置是否有皇后
if(queenPos.contains(pos)){
List<Integer> posList = new ArrayList<Integer>();
posList.add(kx);
posList.add(ky);
ans.add(posList);
break;
}
// 移动到同一方向上的下一个位置
kx += dx;
ky += dy;
}
}
}
return ans;
}
}
复杂度分析:
时间复杂度: O(n + C),其中 n 是数组 queens 的长度,C 是棋盘的大小。
空间复杂度: O(n),即为哈希表所需要的空间。