leetcode 51. N-Queens

题目
那年秋招,遇见这道题,坐着如喽啰。
其实就是dfs+回溯,目的是把所有的情况都遍历一遍。dfs的终止条件是填充到最后一行,因为是一行一行的填充。
然后就是判断在某个位置填充的有效性。分别看这个点的左上方和右上方和上方是否存在Q

class Solution {
    public List<List<String>> solveNQueens(int n) {
        String[][] mat = new String[n][n];
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                mat[i][j] = ".";
            }
        }
        List<List<String>> res = new ArrayList<>();
        dfs(res,mat,0,n);
        return res;
    }
    public void dfs(List<List<String>> res,String[][] mat,int pos,int n){
        if(pos==n-1){
            for(int i=0;i<n;i++){
                if(isValid(mat,pos,i)){
                    mat[pos][i] = "Q";
                    List<String> list = new ArrayList<>();
                    for(int j=0;j<n;j++){
                        String s = "";
                        for(int k=0;k<n;k++){
                            s += mat[j][k];
                        }
                        list.add(s);
                    }
                    res.add(list);
                    mat[pos][i] = ".";
                }
            }
            return;
        }else{
            for(int i=0;i<n;i++){
                if(isValid(mat,pos,i)){
                    mat[pos][i] = "Q";
                    dfs(res,mat,pos+1,n);
                    mat[pos][i] = ".";
                }
            }
        }
    }
    public boolean isValid(String[][] mat,int i,int j){
        for(int row=0;row<mat.length;row++){
            if(mat[row][j].equals("Q")){
                return false;
            }
        }
        int left=j,up=i;
        while(left>=0 && up>=0){
            if(mat[up][left].equals("Q")){
                return false;
            }
            left--;
            up--;
        }
        int right=j,newup=i;
        while(right<mat.length && newup>=0){
            if(mat[newup][right].equals("Q")){
                return false;
            }
            right++;
            newup--;
        }
        return true;
    }
}

写的很丑陋,但是思路基本很清楚。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值