代码随想录-刷题第二十八天

93. 复原 IP 地址

题目链接：93. 复原 IP 地址

思路：切割问题，原理上还是抽象成树形结构，然后使用回溯法。

131 题的要求是：让你把字符串 s 切分成若干个合法的回文串，返回所有的切分方法。

本题的要求是：让你把字符串 s 切分成 4 个合法的 IP 数字，返回所有的切分方法。

所以我们只要把 131 题的解法稍微改改，重写一个 isValid 函数判断合法的 IP 数字，并保证整棵回溯树只有 4 层（即 path 中只有 4 个子串）即可。

public class Solution {
    private List<String> res = new ArrayList<>();
    private List<String> path = new ArrayList<>();

    public List<String> restoreIpAddresses(String s) {
        backtracking(s, 0);
        return res;
    }

    // 回溯算法框架
    private void backtracking(String s, int start) {
        if (start == s.length() && path.size() == 4) {
            // base case，走到叶子节点
            // 即整个 s 被成功分割为合法的四部分，记下答案
            res.add(String.join(".", path));
        }
        for (int i = start; i < s.length(); i++) {
            if (!isValid(s, start, i)) {
                // s[start..i] 不是合法的 ip 数字，不能分割
                continue;
            }
            if (path.size() >= 4) {
                // 已经分解成 4 部分了，不能再分解了
                break;
            }
            // s[start..i] 是一个合法的 ip 数字，可以进行分割
            // 做选择，把 s[start..i] 放入路径列表中
            path.add(s.substring(start, i + 1));
            // 进入回溯树的下一层，继续切分 s[i+1..]
            backtracking(s, i + 1);
            // 撤销选择
            path.remove(path.size() - 1);
        }
    }

    // 判断字符串s在左闭⼜闭区间[start, end]所组成的数字是否合法
    private boolean isValid(String s, int start, int end) {
        // 大于 1 位的时候，不能以 0 开头
        if (s.charAt(start) == '0' && start != end) {
            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. 子集

题目链接：78. 子集

思路：回溯法，首先要将问题抽象成树形结构。注意，组合问题和分割问题都是收集树的叶子节点，而子集问题是找树的所有节点！

class Solution {
    private List<List<Integer>> res = new ArrayList<>();
    private List<Integer> path = new ArrayList<>();

    public List<List<Integer>> subsets(int[] nums) {
        backtracking(nums, 0);
        return res;
    }

    private void backtracking(int[] nums, int start) {
        //「遍历这个树的时候，把所有节点都记录下来，就是要求的子集集合」。
        res.add(new ArrayList<>(path));
        for (int i = start; i < nums.length; i++) {
            path.add(nums[i]);
            backtracking(nums, i + 1);
            path.remove(path.size() - 1);
        }
    }
}

---

90. 子集II

题目链接：90. 子集 II

思路：与上一题思路相同，但是要考虑去重的问题。注意，这里是要在同一个树层上进行去重，而不是在同一个树枝上进行去重。（注意去重需要先对集合排序）

从图中可以看出，同一树层上重复取2 就要过滤掉，同一树枝上就可以重复取2，因为同一树枝上元素的集合才是唯一子集！

class Solution {
    private List<List<Integer>> res = new ArrayList<>();
    private List<Integer> path = new ArrayList<>();

    public List<List<Integer>> subsetsWithDup(int[] nums) {
        // 先排序，让相同的元素靠在一起
        Arrays.sort(nums);
        backtracking(nums, 0);
        return res;
    }

    private void backtracking(int[] nums, int start) {
        res.add(new ArrayList<>(path));
        for (int i = start; i < nums.length; i++) {
            // 跳过当前树层使用过的、相同的元素
            if (i > start && nums[i] == nums[i - 1]) {
                continue;
            }
            path.add(nums[i]);
            backtracking(nums, i + 1);
            path.remove(path.size() - 1);
        }
    }
}

---