代码随想录第24天|回溯

93.复原IP地址

题目链接/文章讲解： 代码随想录

视频讲解：回溯算法如何分割字符串并判断是合法IP？| LeetCode：93.复原IP地址_哔哩哔哩_bilibili

 第一想法

这道题需要穷举所有结果，那么就分割字符串，最后加入"."加入结果集中。

但是在处理过程中发现问题，如何控制分割四次

代码随想录想法

终止条件和131.分割回文串 (opens new window)情况就不同了，本题明确要求只会分成4段，所以不能用切割线切到最后作为终止条件，而是分割的段数作为终止条件。

pointNum表示逗点数量，pointNum为3说明字符串分成了4段了。

然后验证一下第四段是否合法，如果合法就加入到结果集里

代码如下：

if (pointNum == 3) { // 逗点数量为3时，分隔结束
    // 判断第四段子字符串是否合法，如果合法就放进result中
    if (isValid(s, startIndex, s.size() - 1)) {
        result.push_back(s);
    }
    return;
}

• 单层搜索的逻辑

在131.分割回文串 (opens new window)中已经讲过在循环遍历中如何截取子串。

在for (int i = startIndex; i < s.size(); i++)循环中 [startIndex, i] 这个区间就是截取的子串，需要判断这个子串是否合法。

如果合法就在字符串后面加上符号.表示已经分割。

如果不合法就结束本层循环，如图中剪掉的分支：

 代码

class Solution {
    List<String> results = new ArrayList<>();
    public List<String> restoreIpAddresses(String s) {
        getIpAddress(s,0,0);
        return results;
    }
    public void getIpAddress(String s,int startIndex,int pointNum){
        if(pointNum==3){
            if (isValid(s,startIndex,s.length()-1)){
                results.add(s);
            }
            return;
        }
        for(int i = startIndex;i<s.length();i++){
            if(isValid(s,startIndex,i)){
                s=s.substring(0,i+1)+"."+s.substring(i+1);
                pointNum++;
                getIpAddress(s,i+2,pointNum);
                pointNum--;
                s=s.substring(0,i+1)+s.substring(i+2);
            }else{
                break;
            }
        }
    }
    private Boolean isValid(String s, int start, int end) {
        if (start > end) {
            return false;
        }
        if (s.charAt(start) == '0' && start != end) { // 0开头的数字不合法
            return false;
        }
        int num = 0;
        for (int i = start; i <= end; i++) {
            if (s.charAt(i) > '9' || s.charAt(i) < '0') { // 遇到⾮数字字符不合法
                return false;
            }
            num = num * 10 + (s.charAt(i) - '0');
            if (num > 255) { // 如果⼤于255了不合法
                return false;
            }
        }
        return true;
    }
}

78.子集

题目链接/文章讲解：代码随想录

视频讲解：回溯算法解决子集问题，树上节点都是目标集和！ | LeetCode：78.子集_哔哩哔哩_bilibili

第一想法

简单回溯，画图后发现每一步都需要收集。

代码

class Solution {
    List<List<Integer>> results = new ArrayList<>();
    List<Integer> itemPath = new ArrayList<>();
    public List<List<Integer>> subsets(int[] nums) {
        results.add(new ArrayList<>());
        getPath(nums,0);
        return results;
    }
    public void getPath(int[] nums,int startIndex){
        if(startIndex>=nums.length){
            return;
        }
        for(int i = startIndex;i<nums.length;i++){
            itemPath.add(nums[i]);
            results.add(new ArrayList<>(itemPath));
            getPath(nums,i+1);
            itemPath.remove(itemPath.size()-1);
        }
    }
}

 90.子集II

题目链接/文章讲解：代码随想录

视频讲解：回溯算法解决子集问题，如何去重？| LeetCode：90.子集II_哔哩哔哩_bilibili

第一想法

子集和组合问题的组合。数组排序，剪去树层，同时收集结果时间提前。 

代码

class Solution {
    List<List<Integer>> result = new ArrayList<>();
    List<Integer> itemPath = new ArrayList<>();
    public List<List<Integer>> subsetsWithDup(int[] nums) {
        result.add(new ArrayList<>());
        Arrays.sort(nums);
        getPath(nums,0);
        return result;
    }
    public void getPath(int[] nums,int startIndex){
        if(startIndex>=nums.length){
            return;
        }
        for (int i = startIndex; i < nums.length; i++) {
            if(i>startIndex&&nums[i]==nums[i-1]){
                continue;
            }else{
                itemPath.add(nums[i]);
                result.add(new ArrayList<>(itemPath));
                getPath(nums,i+1);
                itemPath.remove(itemPath.size()-1);
            }
        }
    }
}