一文带你领会回溯法

回溯法

概述

回溯法思路的简单描述是：把问题的解空间转化成了图或者树的结构表示，然后使用深度优先搜索策略进行遍历，遍历的过程中记录和寻找所有可行解或者最优解。
基本思想类同于：

• 图的深度优先搜索
• 二叉树的后序遍历

可以得出解决一个回溯问题，实际上就是一个决策树的遍历过程，只有弄清楚以下三个问题

1.路径：也就是已经做出的选择
2.选择列表：当前可以做的选择
3.结束条件：也就是到达决策树底层，无法在做选择的条件

基本解题框架

//定义一个集合存储结果
def backtrack（路径，选择列表）
  if（满足条件）：
     result.add（路径）
     return;
   for 选择 in 选择列表：
       做选择
       backtrack（路径，选择列表）
       撤销选择；

其核心就是在for循环里的递归，在调用递归之前做选择，在递归之后撤销选择

全排列问题入门回溯法

Leecode链接46全排列：https://leetcode-cn.com/problems/permutations/

题目介绍

给定一个 没有重复 数字的序列，返回其所有可能的全排列。

示例:

输入: [1,2,3]
输出:
[
[1,2,3],
[1,3,2],
[2,1,3],
[2,3,1],
[3,1,2],
[3,2,1]
]

解法1

代码

package com.pjh;

import com.sun.xml.internal.fastinfoset.util.ValueArrayResourceException;

import java.util.LinkedList;
import java.util.List;

/**
 * @ClassName: Leetcode46Slution
 * @Author: 86151
 * @Date: 2021/3/31 17:52
 * @Description: TODO
 */
public class Leetcode46Solution {
    List<List<Integer>> result=new LinkedList<List<Integer>>();
    public List<List<Integer>> permute(int[] nums) {
            /*记录路径*/
        LinkedList<Integer> list = new LinkedList<Integer>();
        backtrack(nums,list);
        return result;
    }

    private void backtrack(int[] nums, LinkedList<Integer> list) {
        /*结束条件*/
        if (list.size()==nums.length){
            /*这里要新建一个链表*/
            result.add(new LinkedList(list));
        }
        for (int i = 0; i < nums.length; i++) {
            /*排除不合法的选择*/
            if (list.contains(nums[i])) continue;
            /*做选择*/
            list.add(nums[i]);
            /*进入下层决策树*/
            backtrack(nums,list);
            /*取消选择。即回溯操作*/
            list.removeLast();
        }

    }
}

消耗内存与时间情况

N皇后问题进阶回溯法

Leetcode链接：

题目描述

n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

给你一个整数 n ，返回所有不同的 n 皇后问题 的解决方案。

每一种解法包含一个不同的 n 皇后问题 的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

示例 1：
输入：n = 4
输出：[[".Q…","…Q",“Q…”,"…Q."],["…Q.",“Q…”,"…Q",".Q…"]]

解释：如上图所示，4 皇后问题存在两个不同的解法。
示例 2：
输入：n = 1
输出：[[“Q”]]

提示：

1 <= n <= 9
皇后彼此不能相互攻击，也就是说：任何两个皇后都不能处于同一条横行、纵行或斜线上。

解法1

代码：

package com.pjh;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 * @ClassName: Leetcode51Solution
 * @Author: 86151
 * @Date: 2021/3/31 19:22
 * @Description:  TODO
 */
public class Leetcode51Solution3 {
    /*结果集合*/
  List<List<String>> result=  new ArrayList<List<String>>();
    public List<List<String>> solveNQueens(int n) {
        LinkedList<String> list = new LinkedList<String>();
        backtrack(list,n,0);
        return result;
    }
   /*row当前行数，n为总行数*/
    private void backtrack(LinkedList<String> list, int n,int row) {
        /*结束条件*/
        if (row==n){
            /*在结果集中添加集合*/
            result.add(new LinkedList(list));
            return;
        }
        /*遍历每行的每列*/
        for (int i = 0; i < n; i++) {
            /*判断该节点是否有效是否进行递归*/
            if (isValid(i,list,row,n)){
                /*生成字符串加入到list中*/
                String string = generateString(i, n);
                list.add(string);
                /*进入下一次递归*/
                backtrack(list,n,row+1);
                /*进行回溯，移除之前加入的元素*/
                list.removeLast();
            }
        }

    }

    /*判断该节点是否有效*/
    public boolean isValid(int pos, List<String> list, int row, int n) {
        // 判断竖着有没有重合的
        for(int i = 0; i < row; i++) {
            if(list.get(i).charAt(pos) == 'Q') {
                return false;
            }
        }
        // 判断斜向有没有重合的
        for(int i = row-1, j = pos-1; i >= 0 && j >= 0; i--, j--) {
            if(list.get(i).charAt(j) == 'Q') {
                return false;
            }
        }
        for(int i = row-1, j = pos+1; i >= 0 && j < n; i--, j++) {
            if(list.get(i).charAt(j) == 'Q') {
                return false;
            }
        }
        return true;
    }
     /*生成字符串*/
    public String generateString(int pos, int n) {
        StringBuilder sb = new StringBuilder();
        for(int i = 0; i < pos; i++) {
            sb.append(".");
        }
        sb.append("Q");
        for(int i = pos+1; i < n; i++) {
            sb.append(".");
        }
        return sb.toString();
    }


}

消耗内存与时间情况