用Java代码实现N皇后算法

	/**
     * 初始化数组
     */
    private static int[] queens;

    /**
     * 初始化解
     */
    private static int num;

    /**
     * N皇后问题:  N * N 的棋盘上摆放N个皇后,每一个皇后不能在同一行,同一列,同一斜线
     * 思路: 用一维数组(List),索引表示皇后的行号,对应的值表示皇后的列号
     * 本方法为穷举皇后在第一行各个位置的所有可能性
     * N >= 4
     */
    public static void main(String[] args) {
        solveQueen(5);
    }

    private static void solveQueen(int n) {
        queens = new int[n];
        queen(1, n);
    }

    private static void queen(int j, int n) {
        int i;
        for (i = 1; i <= n; i++) {
            queens[j - 1] = i;
            if (check(j)) {
                // 当最后一行的皇后位置确定, 也就是找到了一组解
                if (j == n) {
                    num = num + 1;
                    System.out.println("解" + num + ": " + Arrays.toString(queens));
                } else {
                    queen(j + 1, n);
                }
            }
        }
    }

    /**
     * 校验之前的所有皇后和当前皇后不在同一列,同一斜线,同一行无需校验
     * @param j 当前皇后索引,也是行数
     * @return boolean
     */
    public static boolean check(int j) {
        int i;
        if (j != 1) {
            for (i = 1; i < j; i++) {
                if ((queens[j - 1] == queens[i - 1]) || Math.abs(j - i) == (Math.abs(queens[j - 1] - queens[i - 1]))) {
                    return false;
                }
            }
        }
        return true;
    }

你可能感兴趣的:(java,算法)