算法笔记|Day19回溯算法

☆☆☆☆☆leetcode 77.组合

题目链接：leetcode 77.组合

题目分析

组合问题是回溯法解决的经典问题，解决这种k层for循环嵌套的问题。每次从集合中选取元素，可选择的范围随着选择的进行而收缩，调整可选择的范围，此题n相当于树的宽度，k相当于树的深度。为了提高效率，可以进行剪枝操作，删去不必要的遍历。

代码

1.未剪枝版本
class Solution {
    List<List<Integer>> res=new ArrayList<>();
    List<Integer> path=new LinkedList<>();

    public List<List<Integer>> combine(int n, int k) {
        backtracking(n,k,1);
        return res;
    }

    public void backtracking(int n,int k,int start){
        if(path.size()==k){
            res.add(new ArrayList(path));
            return;
        }
        for(int i=start;i<=n;i++){
            path.add(i);
            backtracking(n,k,i+1);
            path.removeLast();
        }
    }
}

2.剪枝版本
class Solution {
    List<List<Integer>> res=new ArrayList<>();
    List<Integer> path=new LinkedList<>();

    public List<List<Integer>> combine(int n, int k) {
        backtracking(n,k,1);
        return res;
    }

    public void backtracking(int n,int k,int start){
        if(path.size()==k){
            res.add(new ArrayList(path));
            return;
        }
        for(int i=start;i<=n-(k-path.size())+1;i++){
            path.add(i);
            backtracking(n,k,i+1);
            path.removeLast();
        }
    }
}

3.更好理解的剪枝版本
class Solution {
    List<List<Integer>> res=new ArrayList<>();
    List<Integer> path=new LinkedList<>();

    public List<List<Integer>> combine(int n, int k) {
        backtracking(n,k,1);
        return res;
    }

    public void backtracking(int n,int k,int start){
        if (path.size() > k)
            return;
        if(path.size()==k){
            res.add(new ArrayList(path));
            return;
        }
        for(int i=start;i<=n;i++){
            path.add(i);
            backtracking(n,k,i+1);
            path.removeLast();
        }
    }
}

☆☆☆☆☆leetcode 216.组合总和III

题目链接：leetcode 216.组合总和III

题目分析

此题的终止条件为取到n个元素且和为k，同样为了提高效率，可以进行剪枝操作，删去不必要的遍历。

代码

1.未剪枝版本
class Solution {
    List<List<Integer>> res=new ArrayList<>();
    List<Integer> path=new LinkedList<>();

    public List<List<Integer>> combinationSum3(int k, int n) {
        backtracking(k,n,1,0);
        return res;
    }

    public void backtracking(int k,int n,int start,int sum){
        if(path.size()==k&&sum==n){
            res.add(new ArrayList<>(path));
            return;
        }
        for(int i=start;i<=9;i++){
            sum+=i;
            path.add(i);
            backtracking(k,n,i+1,sum);
            path.removeLast();
            sum-=i;

        }
    }
}

2.剪枝版本
class Solution {
    List<List<Integer>> res=new ArrayList<>();
    List<Integer> path=new LinkedList<>();

    public List<List<Integer>> combinationSum3(int k, int n) {
        backtracking(k,n,1,0);
        return res;
    }

    public void backtracking(int k,int n,int start,int sum){
        if (path.size() > k)
            return;
        if (sum>n)
            return;
        if(path.size()==k&&sum==n){
            res.add(new ArrayList<>(path));
            return;
        }
        for(int i=start;i<=9;i++){
            path.add(i);
            backtracking(k,n,i+1,sum+i);//此处隐藏了sum的回溯
            path.removeLast();
        }
    }
}

提示：
sum+=i;
path.add(i);
backtracking(k,n,i+1,sum);
path.removeLast();
sum-=i;
以上代码可以隐藏sum的回溯，代码改为
path.add(i);
backtracking(k,n,i+1,sum+i);//此处隐藏了sum的回溯
path.removeLast();

☆☆☆☆☆leetcode 17.电话号码的字母组合

题目链接：leetcode 17.电话号码的字母组合

题目分析

此题需要预设每个数字对应的字母，利用回溯法遍历。

代码

class Solution {
    List<String> res=new ArrayList<>();
    StringBuilder s = new StringBuilder();  
    String letterMap[]={"abc","def","ghi","jkl","mno","pqrs","tuv","wxyz"};

    public List<String> letterCombinations(String digits) {
        if(digits.length()==0)
            return res;
        backtracking(digits,0);
        return res;
    }

    public void backtracking(String digits,int index){
        if(index==digits.length()){
            res.add(s.toString());
            return;
        }
        String letter=letterMap[digits.charAt(index)-'2'];
        for(int i=0;i<letter.length();i++){
            s.append(letter.charAt(i));
            backtracking(digits,index+1);
            s.deleteCharAt(s.length()-1);
        }
    }
}

提示：
（内容同算法笔记|Day13二叉树III题目leetcode 257. 二叉树的所有路径的提示）
StringBuilder 是 Java 中的一个可变字符序列类，提供了修改字符串内容的能力，包括追加、插入和删除等操作，且这些操作都是在原对象上进行的，不会生成新的对象，因此在进行大量字符串操作时，使用 StringBuilder 可以显著提高性能。
补充StringBuilder 的常用方法：
StringBuilder()：构造一个空的 StringBuilder
StringBuilder(CharSequence seq)：构造一个包含指定字符序列的 StringBuilder
StringBuilder(int capacity)：构造一个具有指定初始容量的 StringBuilder
append(…)：向字符序列的末尾追加指定的内容（可以是各种类型的数据，如 String、char、int 等），并返回 this 对象的引用
insert(int offset, CharSequence csq)：在指定位置插入指定的字符序列
delete(int start, int end)：删除此序列的子字符串中的字符
replace(int start, int end, String str)：用指定字符串替换此序列的子字符串
length()：返回此字符序列的长度
toString()：返回此序列中数据的字符串表示形式