N皇后问题:通过DFS方式输出所有的棋子放置方案

3 篇文章 0 订阅

https://leetcode-cn.com/problems/n-queens/

N皇后问题是指在N×N格的国际象棋上摆放八个皇后,使其不能互相攻击,即任意两个皇后都不能处于同一行、同一列或同一斜线上,问有多少种摆法。

思路

1.如何判断(row, col)位置是否能放皇后
(1)使用col存储已经摆放皇后的列索引
(2)sum存储 / 方向的对角线,因为 / 方向的对角线的坐标满足 (x1+y1) = (x2+y2),例如  (0+3)=(1+2)=(2+1)=(3+0)
(3)diff存储 \ 方向的对角线,因为 \ 方向的对角线的坐标满足 (x1-y1) = (x2-y2),例如  (0-0)=(1-1)=(2-2)=(3-3)

2. dfs :遍历第row行的每一列元素。
(1)如果发现能放皇后,则col、sum、diff都加上当前元素( i )的相关值,然后构造出该行的棋盘。继续遍历下一行(row+1),摆放皇后棋。递归结束后,当前row行的放置回退一步(col、sum、diff,result 都移除刚刚加入的元素),继续遍历下一列的位置( i++ )。
(3)直到row==n,则找到一种棋子放置方案。
(4)如果第row行中所有棋子都不能放,当前的dfs达到i=n时,当前遍历结束,返回到上一层继续判断下一列的位置。

class Solution {
    private List<List<String>> ans = new ArrayList<>();
    private Set<Integer> col = new HashSet<>();
    private Set<Integer> sum = new HashSet<>();
    private Set<Integer> diff = new HashSet<>();
    private int n = 0;
    
    public List<List<String>> solveNQueens(int n) {
        this.n = n;
        dfs(0,new ArrayList<>());
        return ans;
    }
    
    void dfs(int row, List<String> result){
        if(row == n){
            ans.add(new ArrayList<>(result));
            return;
        }
        
        for(int i = 0;i<n;i++){
            if(!col.contains(i) && !sum.contains(row+i) && !diff.contains(row-i)){
                col.add(i);
                sum.add(row+i);
                diff.add(row-i);
                
                String tmp = "";
                for(int j = 0;j<n;j++){
                    tmp = tmp + (i==j?"Q":".");
                }
                result.add(tmp);
                
                dfs(row+1,result);
                
                col.remove((Integer)i);
                sum.remove((Integer)(row+i));
                diff.remove((Integer)(row-i));
                result.remove(result.size() - 1);
            }
        }
    }
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值