使用回溯算法求解N皇后问题

N皇后问题描述

在一个N * N的国际象棋棋盘上需要放置N个皇后,放置时要保证这些皇后不能攻击彼此,也就是要保证在同一行、同一列、同一斜线上都只有一个皇后。该问题要求求出所有满足要求的放置方案。

八皇后

回溯算法简介

回溯算法对问题的要求

当一个问题满足:

  1. 多步:需要多步来完成
  2. 多选:每一步有多种决策选择,并且你无法预见哪个选择会使你得到解,哪个选择会使你得不到解
  3. 瞻前:后续步骤会依赖于先前步骤的选择

这个问题就可以用回溯算法来解决。

拿求解迷宫路径的问题进行比对:

  • 要求1:寻找迷宫路径时,每一步操作就确定路径上的一个点,需要多步来确定一条完整的路径
  • 要求2:当你在当前步骤确定路径的下一个节点时,你可能有很多节点可选,并且你不知道选哪个节点才是可以求得最终解的
  • 要求3:你在当前步骤的候选节点是由你之前步骤确定的路径决定的

所以求解迷宫问题可以通过回溯算法来解决。读者可以怎么样可以让N皇后问题也符合回溯算法的要求,没有思路可以看下文“使用回溯算法求解N皇后问题”的部分

回溯算法的思想

回溯算法用到的思想是:从一条路往前走,能进则进,不能进则退回来,换一条路再试,以此来把所有的可能性都考虑一遍。总结起来也就是深度优先搜索。

回溯算法的模板

回溯算法是由递归函数实现的,这个递归函数的基本骨架如下:

void backtrack(resultStack, arg1, arg2, ...){
    // 当满足题目的某种要求时,说明此时找到了问题的一个解
    if (some condition){
        // 在这里根据题目要求做出相关操作
        // 这里的操作一般是将resultStack复制一份,加入到所有解的集合中
        some operations;
        
        // 直接返回
        return;
    }
    
    // candidates 表示"要求2"中提到的多种候选决策
    // 在这里通过for循环遍历所有的候选决策
    for (candidate : candidates){
        // 当候选条件不符合要求时,跳过这个条件
        if (candidate does not satisfy some requirements){
            continue;
        }
        
        // 下面三行是这个函数的核心,体现了回溯算法能进则进的思想
        // 先把当前候选条件加入结果栈中,然后进入下一步,
        // 从下一步中返回后,将当前候选条件出栈,把结果栈恢复到之前的状态,
        // 以便考虑下一个candidate(for循环的下一个元素)
        resultStack.push(candidate);
        backtrack(resultStack, arg1, arg2);
        resultStack.pop();
    }
}

使用回溯算法求解N皇后问题

题目要求请看这里: https://leetcode-cn.com/problems/n-queens/

如何确定思路呢?可以从回溯算法的要求出发:多步、多选、瞻前。首先要把问题分解成一个多步可解的情形,我们可以这样想:棋盘的每行只能而且必须放一个皇后,只要把所有行都放上一个皇后,放置方案就出来了,所以此时回溯算法的每一步就是:在当前行中放置一个皇后。然后我们就可以看这样满不满足回溯算法的要求2和要求3了,每一行有N列,每一列都是一个决策选择,所以满足要求2;在选择当前行要在哪一列放置皇后时,要保证该位置不会和之前放置的皇后在同一列、或者同一斜线上,所以满足要求3。

这样一来解决方案其实很简单了,只要套上面的模板就可以了:

import java.util.*;

class Solution {
    // 用二维数组表示棋盘
    private char[][] board;
    // N皇后问题的N
    private int n;
    // 所有可行的放置方案的集合
    private List<List<String>> solutions = new ArrayList<>();
    
    // 主函数
    public List<List<String>> solveNQueens(int n) {
        this.n = n;
        // 初始化棋盘,并且把每一个单元格都设置为'.',表示棋盘是空的,没有皇后
        this.board = new char[n][n];
        for (int i = 0; i < n; i++){
            for (int j = 0; j < n; j++){
                board[i][j] = '.';
            }
        }
        
        // 进入回溯函数
        backtrack(new ArrayList<>(), new ArrayList<>(), new ArrayList<>(), 0);
        return solutions;
    }
    
    // 实现回溯算法的函数
    // usedCols 表示使用过的列号,列号从0开始
    // usedDianonals1 表示使用过的、方向为从左上到右下的 斜线号
    // usedDianonals2 表示使用过的、方向为从右上到左下的 斜线号
    // row 表示当前步骤的行号,行号从0开始
    private void backtrack(List<Integer> usedCols, List<Integer> usedDiagonals1, List<Integer> usedDiagonals2, int row){
        // 行号从0开始,所以棋盘的最后一行的标号是n-1,
        // 当行号为n时,说明所有的行都已经放置了正确位置的皇后,可以添加解了。
        if (row == n){
            addSolution();
            return;
        }
        
        // 遍历当前行的所有列
        for (int j = 0; j < n; j++){
            if (usedCols.contains(j) || usedDiagonals1.contains(row - j) || usedDiagonals2.contains(row + j)){
                continue;
            }
            
            board[row][j] = 'Q';
            usedCols.add(j);
            usedDiagonals1.add(row - j);
            usedDiagonals2.add(row + j);
            backtrack(usedCols, usedDiagonals1, usedDiagonals2, row + 1);
            board[row][j] = '.';
            usedCols.remove(usedCols.size() - 1);
            usedDiagonals1.remove(usedDiagonals1.size() - 1);
            usedDiagonals2.remove(usedDiagonals2.size() - 1);
        } 
        
    }
    
    private void addSolution(){
        List<String> solution = new ArrayList<>();
        for (int i = 0; i < n; i++){
            StringBuilder builder = new StringBuilder();
            for (int j = 0; j < n; j++){
                builder.append(board[i][j]);
            }
            solution.add(builder.toString());
        }
        
        solutions.add(solution);
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,657评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,662评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,143评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,732评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,837评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,036评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,126评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,868评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,315评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,641评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,773评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,470评论 4 333
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,126评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,859评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,095评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,584评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,676评论 2 351