回溯法

Posted 10zhang

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了回溯法相关的知识,希望对你有一定的参考价值。

回溯法

全排列系列

46题:

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

示例:
输入: [1,2,3]
输出:
[
  [1,2,3],
  [1,3,2],
  [2,1,3],
  [2,3,1],
  [3,1,2],
  [3,2,1]
]
代码:
public List<List<Integer>> permute(int[] nums) {
   List<List<Integer>> list = new ArrayList<>();
   // Arrays.sort(nums); // 不必先排序
   backtrack(list, new ArrayList<>(), nums);
   return list;
}

private void backtrack(List<List<Integer>> list, List<Integer> tempList, int [] nums){
   if(tempList.size() == nums.length){
      list.add(new ArrayList<>(tempList));
   } else{
      for(int i = 0; i < nums.length; i++){ 
         if(tempList.contains(nums[i])) continue; // 元素已经存在,跳过
         tempList.add(nums[i]);
         backtrack(list, tempList, nums);
         tempList.remove(tempList.size() - 1);
      }
   }
} 

47题:

给定一个可包含重复数字的序列,返回所有不重复的全排列。

示例:
输入: [1,1,2]
输出:
[
  [1,1,2],
  [1,2,1],
  [2,1,1]
]
代码:
 public List<List<Integer>> permuteUnique(int[] nums) {
        List<List<Integer>> res=new ArrayList<>();
        List<Integer> temp=new ArrayList<>();
        boolean[] used=new boolean[nums.length]; //指示该值是否已经添加到列表中
        Arrays.sort(nums);     //对数组排序,确保可以跳过相同的值
        helper(res,temp,used,nums);
        return res;

    }
    public void helper(List<List<Integer>> res,List<Integer> temp,boolean[] used,int[] nums){
        if (temp.size()==nums.length){
            res.add(new ArrayList<>(temp));
        }else {
            for (int i = 0; i <nums.length ; i++) {
                //列表中已经添加过这个位置的值,跳过
                if (used[i]) continue;
                //当一个数字与之前的数字具有相同的值时,我们只有在使用前一个数字时才能使用此数字
                if (i>0&&nums[i]==nums[i-1]&&!used[i-1]) continue;  
                
                used[i]=true;
                temp.add(nums[i]);
                helper(res,temp,used,nums);
                used[i]=false;
                temp.remove(temp.size()-1);
            }
        }
    }

子集系列

78题:

给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集。 说明:解集不能包含重复的子集。

示例:
输入: nums = [1,2,3]
输出:
[
  [3],
  [1],
  [2],
  [1,2,3],
  [1,3],
  [2,3],
  [1,2],
  []
]
代码:
public List<List<Integer>> subsets(int[] nums) {
    List<List<Integer>> list = new ArrayList<>();
   // Arrays.sort(nums);//不必要
    backtrack(list, new ArrayList<>(), nums, 0);
    return list;
}

private void backtrack(List<List<Integer>> list , List<Integer> tempList, int [] nums, int start){
    //先存结果,递归边界不用显式确定,如果无法添加自然不会再递归
    list.add(new ArrayList<>(tempList));
    for(int i = start; i < nums.length; i++){
        tempList.add(nums[i]);
        backtrack(list, tempList, nums, i + 1);
        tempList.remove(tempList.size() - 1);
    }
}

另一种迭代方法

代码:
public List<List<Integer>> subsets(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        result.add(new ArrayList<>());
        for(int n : nums){
            int size = result.size();
            for(int i=0; i<size; i++){
                List<Integer> subset = new ArrayList<>(result.get(i));
                subset.add(n);
                result.add(subset);
            }
        }
        return result;
    }
解释:

在迭代所有数字时,对于每个新数字,我们可以选择它,也可以不选择它
1,如果选择,只需将当前编号添加到每个现有子集。
2,如果没有选择,只保留所有现有的子集。
我们只是将两者结合起来。

例如,{1,2,3}在内部我们有一个结果集[[]]
考虑1,如果不使用它,仍然[],如果使用1,将它添加到[],所以我们现在有[1]
结合它们,现在我们有[[],[1]]作为所有可能的子集

接下来考虑2,如果不使用它,我们仍然有[[],[1]],如果使用2,只需在每个前面的子集中加2,我们有[2],[1,2]
结合他们,现在我们有[[],[1],[2],[1,2]]

接下来考虑3,如果不使用它,我们仍然有[[],[1],[2],[1,2]],如果使用3,只需在每个前面的子集中加3,我们[[3], [1,3],[2,3],[1,2,3]]
结合它们,现在我们有[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]

90题:

给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。说明:解集不能包含重复的子集。

示例:

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

代码:

public List<List<Integer>> subsetsWithDup(int[] nums) {
    List<List<Integer>> list = new ArrayList<>();
    Arrays.sort(nums);//排序必要,跳过重复
    backtrack(list, new ArrayList<>(), nums, 0);
    return list;
}

private void backtrack(List<List<Integer>> list, List<Integer> tempList, int [] nums, int start){
    list.add(new ArrayList<>(tempList));
    for(int i = start; i < nums.length; i++){
        if(i > start && nums[i] == nums[i-1]) continue; // 跳过重复元素(剪枝)
        tempList.add(nums[i]);
        backtrack(list, tempList, nums, i + 1);
        tempList.remove(tempList.size() - 1);
    }
} 

另一种迭代方法:(在78题迭代法上改进)

代码:
public List<List<Integer>> subsetsWithDup(int[] nums) {
        Arrays.sort(nums);  //不要忘了排序
        List<List<Integer>> result = new ArrayList<>();
        result.add(new ArrayList<>());
        int size=0;
        for(int j=0;j<nums.length;j++){
            
            int start =(j>=1&&nums[j]==nums[j-1])?size:0;  //定起始位置,这里的size还没更新,所以是上一次迭代后的结果数目
            size=result.size();  //size用来保存当前结果数目
            for(int i=start; i<size; i++){
                List<Integer> subset = new ArrayList<>(result.get(i));
                subset.add(nums[j]);
                result.add(subset);
            }
        }
        return result;
    }

组合系列

组合 77题:给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。

示例:

输入: n = 4, k = 2
输出:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

代码:

public List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> res=new ArrayList<>();
        helper(res, new ArrayList<Integer>(),1,n,k);
        return res;

    }
    private void helper(List<List<Integer>> res,List<Integer> templist,int start,int n,int k){
        if (k==0){
            res.add(new ArrayList<>(templist));
            return;
        }
        for (int i=start;i<=n;i++){
            templist.add(i);
            helper(res,templist,i+1,n,k-1);
            templist.remove(templist.size()-1);
        }
    }

组合总和系列:

组合总和1:39题:

给定一个无重复元素的数组** candidates** 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的数字可以无限制重复被选取

说明:所有数字(包括 target)都是正整数。解集不能包含重复的组合。

示例:
输入: candidates = [2,3,5], target = 8,
所求解集为:
[
  [2,2,2,2],
  [2,3,3],
  [3,5]
]
代码:
public List<List<Integer>> combinationSum(int[] candidates, int target) {
        Arrays.sort(candidates);       //先排序便于跳过大于target的数字
        List<List<Integer>> res=new ArrayList<>();      
        getRes(res,new ArrayList<Integer>(),candidates,target,0);
        return res;

    }
    private void getRes(List<List<Integer>> res, List<Integer> cur,int[] candidates,int target,int start){
        if (target>0){
            for (int i=start;i<candidates.length&&target>=candidates[i];i++){     //从start开始往后找
                cur.add(candidates[i]);
                getRes(res,cur,candidates,target-candidates[i],i);
                cur.remove(cur.size()-1);  //调用返回后及时清除
            }
        }else if (target==0){
            res.add(new ArrayList<>(cur));  //此处不能res.add(cur) 只能添加cur的副本,不然对cur的remove会改变res中相应添加的项
        }
    }

组合总和2 :40题:

题目同1,但是candidates 中的每个数字在每个组合中只能使用一次。

示例:
输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
  [1, 7],
  [1, 2, 5],
  [2, 6],
  [1, 1, 6]
]
代码:
public List<List<Integer>> combinationSum2(int[] candidates, int target) {
        Arrays.sort(candidates);
        List<List<Integer>> res=new ArrayList<>();
        getRes(res,new ArrayList<Integer>(),candidates,target,0);
        return res;

    }
    private void getRes(List<List<Integer>> res, List<Integer> cur,int[] candidates,int target,int start){
        if (target>0){
            for (int i=start;i<candidates.length&&target>=candidates[i];i++){
                if (i>start&&candidates[i]==candidates[i-1]){ continue;}  //跳过数组中的重复元素(剪枝) 注意从当前start开始
                cur.add(candidates[i]);
                getRes(res,cur,candidates,target-candidates[i],i+1);   //注意是i+1 不能重复使用数组中的元素
                cur.remove(cur.size()-1);
            }
        }else if (target==0){
            res.add(new ArrayList<>(cur));
        }
    }

组合总和3:216题:

找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。

示例:
输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
代码:
public List<List<Integer>> combinationSum3(int k, int n) {
    int[] num = {1,2,3,4,5,6,7,8,9};
    List<List<Integer>> result = new ArrayList<List<Integer>>();
    helper(result, new ArrayList<Integer>(), num, k, n,0);
    return result;
    }

public void helper(List<List<Integer>> result, List<Integer> list, int[] num, int k, int target, int start){
    if (k == 0 && target == 0){
        result.add(new ArrayList<Integer>(list));
    } else {
        for (int i = start; i < num.length && target > 0 && k >0; i++){
            list.add(num[i]);
            helper(result, list, num, k-1,target-num[i],i+1);
            list.remove(list.size()-1);
        }
    }
}

涉及字符串的回溯问题

括号生成 22题:

给出 n 代表生成括号的对数,请你写出一个函数,使其能够生成所有可能的并且有效的括号组合。例如,给出 n = 3,生成结果为:

[
  "((()))",
  "(()())",
  "(())()",
  "()(())",
  "()()()"
]

思路:最初的想法是生成所有的组合情况,再写辅助函数判断是否合法。但这么做不必要,因为给定了括号的对数,只要最后生成的左右括号数都等于括号对数就是合法。可以在递归参数中引入左右括号各自的计数器,来记录数量,在当前Stirng中字符数量达到2*括号对数时,添加结果到list。

代码:

 public List<String> generateParenthesis(int n) {
        List<String> l=new ArrayList<>();
        backcrack(l,"",0,0,n);
        return l;

    }
    public void backcrack(List<String> l,String current,int open,int close,int max){
        if (current.length()==2*max){
            l.add(current);
        }else {
            if (open<max){   //max为括号对数
                backcrack(l,current+‘(‘,open+1,close,max);
            }
            if (close<open){//注意这里 只有右括号的数量小于左括号的数量,才可以加右括号
                backcrack(l,current+‘)‘,open,close+1,max);
            }
        }
    }
    //只生成合法的情况。这里的current相当于数组回溯问题中的List<Integer> templist

注意!:java里边String对象是不可变的,也就是说current+‘(‘不是简单的在原来的current指向的String对象后面加上‘(‘,而是又新生成了一个current+‘(‘,和原来的不是一个了。

电话号码的字母组合 17题:

给定一个仅包含数字 2-9 的字符串,返回所有它能表示的字母组合。给出数字到字母的映射如下(与电话按键相同)。注意 1 不对应任何字母。

技术分享图片

示例:

输入:"23"
输出:["ad", "ae", "af", "bd", "be", "bf", "cd", "ce", "cf"].

代码:

private static final String[] KEYS = { "", "", "abc", "def", "ghi", "jkl", "mno", "pqrs", "tuv", "wxyz" };
    
        public List<String> letterCombinations(String digits) {
           
            List<String> ret = new LinkedList<String>();
             if(digits == null || digits.length() == 0) return ret;
            combination("", digits, 0, ret);
            return ret;
        }
    
        private void combination(String prefix, String digits, int offset, List<String> ret) {
            if (offset >= digits.length()) {
                ret.add(prefix);
                return;
            }
            String letters = KEYS[(digits.charAt(offset) - ‘0‘)];
            for (int i = 0; i < letters.length(); i++) {
                combination(prefix + letters.charAt(i), digits, offset + 1, ret);
            }
        }
        //这里定义的前缀prefix相当于数组中回溯问题的List<Integer> templist,都是作为递归函数的参数保存生成结果的。

复原IP地址 93题:

给定一个只包含数字的字符串,复原它并返回所有可能的 IP 地址格式。

示例:

输入: "25525511135"
输出: ["255.255.11.135", "255.255.111.35"]

注意:

什么是有效的IP地址格式?

代码:

public List<String> restoreIpAddresses(String s) {
        List<String> res = new ArrayList<>();
        helper(s,"",res,0);
        return res;
    }
    public void helper(String s, String tmp, List<String> res,int n){
        if(n==4){
            if(s.length()==0) res.add(tmp.substring(0,tmp.length()-1));
            //substring here to get rid of last ‘.‘
            return;
        }
        for(int k=1;k<=3;k++){
            if(s.length()<k) continue;  //剪枝
            int val = Integer.parseInt(s.substring(0,k));
            if(val>255 || k!=String.valueOf(val).length()) continue;
            /*in the case 010 the parseInt will return len=2 where val=10, but k=3, skip this.*/
            helper(s.substring(k),tmp+s.substring(0,k)+".",res,n+1);
            //每次递归,s都是传的k起始的一个子串,这样传参更简便,不用麻烦地传s下标
        }
    }

分割回文串 131题:

给定一个字符串 s,将 s 分割成一些子串,使每个子串都是回文串。返回 s 所有可能的分割方案。

示例:

输入: "aab"
输出:
[
  ["aa","b"],
  ["a","a","b"]
]

代码:

   public List<List<String>> partition(String s) {
        List<List<String>> res=new ArrayList<>();
        List<String> cur=new ArrayList<>();
        helper(res,s,cur,0,s.length());
        return res;
    }
    //判断回文字符串的辅助函数
    public boolean isPalindrome(String s){
        int n=s.length()-1,i=0;
        while (i<n){
            if (s.charAt(i)!=s.charAt(n)){
                return false;
            }
            i++;
            n--;
        }
        return true;

    }
    
    //回溯函数
    public List<List<String>> helper(List<List<String>> res, String s,List<String> cur,int start,int len){
        if (start==len){
            res.add(new ArrayList<>(cur));
        }
        for (int i=start+1;i<=len;i++){
           if (isPalindrome(s.substring(start,i))){
               cur.add(s.substring(start,i));
               helper(res,s,cur,i,len);
               cur.remove(cur.size()-1);
           }
        }
        return res;
    }

图解:

技术分享图片

其他问题:

单词搜索 79题:

给定一个二维网格和一个单词,找出该单词是否存在于网格中。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。

示例:

board =
[
  [‘A‘,‘B‘,‘C‘,‘E‘],
  [‘S‘,‘F‘,‘C‘,‘S‘],
  [‘A‘,‘D‘,‘E‘,‘E‘]
]

给定 word = "ABCCED", 返回 true.
给定 word = "SEE", 返回 true.
给定 word = "ABCB", 返回 false.

分析:实际上这道题属于DFS 在一个二维数组里面做查找

代码:

static boolean[][] visited;
    public boolean exist(char[][] board, String word) {
        visited = new boolean[board.length][board[0].length];
        
        for(int i = 0; i < board.length; i++){
            for(int j = 0; j < board[i].length; j++){
                if((word.charAt(0) == board[i][j]) && search(board, word, i, j, 0)){
                    return true;
                }
            }
        }
        
        return false;
    }
    
    private boolean search(char[][]board, String word, int i, int j, int index){
        if(index == word.length()){
            return true;     //查找到最后 说明找到 返回
        }
        
        if(i >= board.length || i < 0 || j >= board[i].length || j < 0 || board[i][j] != word.charAt(index) || visited[i][j]){
        //下标越界,当前搜索位置的字母与目标字母不同,当前位置字母已经访问过 这些情况都属于没找到,返回flase
            return false;
        }
        
        visited[i][j] = true;    //置为已访问
        if(search(board, word, i-1, j, index+1) || 
           search(board, word, i+1, j, index+1) ||
           search(board, word, i, j-1, index+1) || 
           search(board, word, i, j+1, index+1)){
            return true;   //有一路找到就直接返回就行 最终一定是找到了 就不用管visited是不是置回false了
        }
        
        visited[i][j] = false;    //递归回溯后记得再置回未访问 以便再找另一路
        return false;  //进行到这一步还没return 这一路是最终没找到
    }

格雷编码 89题:

格雷编码是一个二进制数字系统,在该系统中,两个连续的数值仅有一个位数的差异。
给定一个代表编码总位数的非负整数 n,打印其格雷编码序列。格雷编码序列必须以 0 开头。

示例:

输入: 2
输出: [0,1,3,2]
解释:
00 - 0
01 - 1
11 - 3
10 - 2

对于给定的 n,其格雷编码序列并不唯一。
例如,[0,2,3,1] 也是一个有效的格雷编码序列。

00 - 0
10 - 2
11 - 3
01 - 1
输入: 0
输出: [0]
解释: 我们定义格雷编码序列必须以 0 开头。
     给定编码总位数为 n 的格雷编码序列,其长度为 2n。当 n = 0 时,长度为 20 = 1。
     因此,当 n = 0 时,其格雷编码序列为 [0]。

分析:这题标签是回溯法,但实际上回溯不是最好的办法。

代码:

public List<Integer> grayCode(int n) {
    List<Integer> rs=new ArrayList<Integer>();
    rs.add(0);
    for(int i=0;i<n;i++){
        int size=rs.size();
        for(int k=size-1;k>=0;k--)
            rs.add(rs.get(k) | 1<<i);  //或者写成 rs.add(rs.get(k)+(1<<i))
    }
    return rs;
}









以上是关于回溯法的主要内容,如果未能解决你的问题,请参考以下文章

用回溯法求01背包问题,怎样使用C++模板啊,迫切求指点!

用回溯法做0-1背包问题,这两行(程序中标注)是干嘛?为啥又要减?

回溯法八皇后问题(递归和非递归)

暴力穷举和回溯法(八皇后问题)

0-1背包问题如下,画用回溯法求解时的搜索情况,急用啊

[XJTUSE 算法设计与分析] 第五章 回溯法