天天看点

DFS实现N皇后问题

给定一个N x N 的空格子 。。。。。对于n皇后问题是什么问题请自行百度。。。话不多说,见代码

最终是用一个List来保存n皇后的输出图像

#输入为n,输出为合法的n皇后表格图像
  public List<List<String>> solveNQueens(int n) {
        
        List<List<String>> list=new ArrayList<>();
        char[][] c=new char[n][n];
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                c[i][j]='.';
        
        dfs(c,list,0);
        return list;
        
    }
    

    public void dfs(char[][] c,List<List<String>> list,int colIndex){
        if(colIndex==c.length){
            list.add(getList(c));
        }
        
        for(int i=0;i<c.length;i++){
            if(isValidate(c,i,colIndex)){
                c[i][colIndex]='Q';
                dfs(c,list,colIndex+1);
                c[i][colIndex]='.';
            }
        }
    }
    
    public boolean isValidate(char[][] c,int x,int y){
        for(int i=0;i<c.length;i++){
            for(int j=0;j<y;j++){
                if(c[i][j]=='Q' && (i==x || x+j==y+i || x+y== i+j))
                    return false;
            }
        }
        return true;
    }
    
    public List<String> getList(char[][] c){
        List<String> result=new ArrayList<String>();
        for(int i=0;i<c.length;i++){
            String cur=new String(c[i]);
            result.add(cur);
        }
        return result;
    }