N皇后 java

代码

class Solution {
    public static List<List<String>> output;
    public List<List<String>> solveNQueens(int n) {
        output = new ArrayList<>();
        // 声明一个长度为n的数组用来代表 第n行棋子是在第result[n]列
        int[] result = new int [n];
        calnQueens(0, n, result);
        return output;
    }
    
    // n 皇后问题 row代表计算到了到第row行
    private static void calnQueens(int row, int n, int[] result){
        if (row == n){
            // 到达第n行代表已经得到一个将解决方案 直接返回即可
            // 根据result数组将结果加入到output列表中
            getPrint(result);
            return;
        }
        // 若不是第n行 则说明需要继续判断该行棋子应该在那一列
        for (int column = 0; column < n; column++){
            // 判断第row行 放置在column列的棋子是否满足要求
            if (isOK(row, column, result)){
                result[row] = column;
                // 递归判断下一行的情况
                calnQueens(row + 1, n, result);
            }
            // 不满足要求 回溯下一列 对应操作column++
        }
    }
    
    // row代表行数 column代表列数 result代表满足规则的棋子在第n行中的位置
    private static boolean isOK(int row, int column, int[] result){
        // 判断棋子的位置是否正确 不正确返回false
        for (int i = 0; i < row; i++){
            // 第一个条件排除的是相同列的问题
            // 第二个条件排除的是对角线列的左下角（这个地方错了，column - result[i] = -(row-i),斜率为-1呀，是右下角才对）
            // 第三个条件排除的是对角线列的右下角
            if (column == result[i] || column == result[i] - row + i || column == result[i] + row - i){
                return false;
            }
        }
        return true;
    }
    
    private static void getPrint(int[] result){
        List<String> one = new ArrayList<>();
        for (int row = 0; row < result.length; row++){
            // 一行一个StringBuilder
            StringBuilder str = new StringBuilder();
            for (int column = 0; column < result.length; column++){
                if (column == result[row]){
                    str.append("Q");
                }else{
                    str.append(".");
                }
            }
            one.add(str.toString()); 
        }
        output.add(one);
    }
}

我提交的解法

package com.daleyzou.leetcode;

import java.lang.management.MemoryType;
import java.util.ArrayList;
import java.util.List;

/**
 * LeetCode51
 * @description https://www.cnblogs.com/daleyzou/p/NQueen.html
 * @author daleyzou
 * @date 2019年12月25日 12:45
 * @version 3.1.0
 */
public class LeetCode51 {
    List<List<String>> resultList = new ArrayList<>();
    public List<List<String>> solveNQueens(int n) {

        int[] result = new int[n];
        backTrack(0, n, result);
        return resultList;
    }

    private void backTrack(int row, int n, int[] result) {
        if (row == n){
            printResult(result);
            return;
        }
        for (int column = 0; column < n; column++){
            if (isOK(row, column, result)){
                result[row] = column;
                backTrack(row + 1, n, result);
            }
        }

    }

    private boolean isOK(int row, int column, int[] result) {
        for (int i = 0; i < row; i++){
            if (column == result[i] || Math.abs(column - result[i]) == Math.abs(row - i)){
                return false;
            }
        }
        return true;
    }

    private void printResult(int[] result) {
        List<String> item = new ArrayList<>();
        for (int i = 0; i < result.length; i++){
            StringBuilder sb = new StringBuilder();
            for (int j = 0; j < result.length; j++){
                if (j == result[i]){
                    sb.append("Q");
                }else {
                    sb.append(".");
                }
            }
            item.add(sb.toString());
        }
        resultList.add(item);
    }
}