lc_other_1222_queensAttacktheKing


题目: 可以攻击国王的皇后 middle

在一个 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] 的皇后无法攻击到国王,因为她和国王不在同一行/列/对角线上。


思路:
    使用direction[][]代表8个方向,一次for循环寻找;不要使用8次for循环,每次for循环都大同小异

 

 

 

package leetCode.Other;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class lc_other_1222_queensAttacktheKing {

    public List> queensAttacktheKing(int[][] queens, int[] king) {
        List> ans = new ArrayList<>();
        if (queens == null || queens.length == 0 || king == null)
            return ans;
        boolean[][] chess = new boolean[8][8];
//        将有皇后的位子设为true,这样下面就直接和chess比较而不需要遍历8次queen数组
        for (int[] queen : queens) {
            chess[queen[0]][queen[1]] = true;
        }
//        8个方向
        int[][] direc = {{-1, -1}, {-1, 0}, {-1, 1}, {0, 1}, {1, 1}, {1, 0}, {1, -1}, {0, -1}};
//      循环8个方向,只要碰到第一个Queen,立刻停止该方向的搜索,继续下一个方向的搜索
        for (int n = 0; n < direc.length; n++) {
            for (int i = king[0], j = king[1]; condition(i, j); i += direc[n][0], j += direc[n][1]) {
                if (chess[i][j]) {
                    /*
                    ArrayList list = new ArrayList<>();
                    list.add(i);
                    list.add(j);*/
                    ans.add(Arrays.asList(i, j));//这个方法很好Arrays.asList(T...a)
                    break;//一定要break,只保存碰到的第一个queen
                }
            }
        }

        return ans;
    }

    //将条件变成方法方便查看
    public boolean condition(int i, int j) {
        if (i >= 0 && i < 8 && j >= 0 && j < 8)
            return true;
        return false;

    }
}

 

你可能感兴趣的:(LeetCode)