leetcode hard模式专杀之51. N-Queens_leetcode hard 八皇后-优快云博客

本文深入探讨了经典的八皇后问题，并提供了一种通用解决方案。通过使用回溯法，文章详细介绍了如何编写代码来解决N皇后问题，包括关键的边界条件处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

继续leetcode hard模式，八皇后问题的一般版，这种举一反三的题目值得好好琢磨，典型的回溯法，思路就不多说了，上代码，注意一一些边界条件的细节：

public class Solution {
    public List<String>  transfer(int[] q){
        List<String> result = new ArrayList<>();
        for(int i = 0; i<q.length; i++){
            StringBuilder sb = new StringBuilder();
            for(int j=0; j<q.length; j++){
                if(j==q[i]){
                    sb.append("Q");
                }else{
                    sb.append(".");
                }
            }
            result.add(sb.toString());
        }
        return result;
    }
    public List<List<String>> solveNQueens(int n) {
        List<List<String>> result = new ArrayList<>();
        int[] q=new int[n];
        for(int i = 0 ; i<n; i++){
            enumerate(result, i, q, 0);
        }

        return result;
    }

    public boolean isConsistent(int[] q, int line, int index){
        if(line==0){
            return true;
        }else{
            for(int i=0; i<line;i++){
                //i行q[i]列 vs line行index列
                if(q[i]==index){
                    //同列
                    return false;
                }
                if(index-q[i]==line-i){
                    return false;
                }
                if(index-q[i]==i-line){
                    return false;
                }
            }
            return true;
        }
    }

    public void enumerate(List<List<String>> result, int index, int[] q, int line){
        if(line==q.length-1){
            if(isConsistent(q, line, index)){
                q[line] = index;
                result.add(transfer(q));
            }
        }else if(line == 0){
            q[line] = index;
            for(int j = 0; j<q.length; j++){
                enumerate(result, j, q, line+1);
            }
        }else{
            if(isConsistent(q, line, index)){
                q[line] = index;
                for(int j = 0; j<q.length; j++){
                    enumerate(result, j, q, line+1);
                }
            }
        }
    }
}