代码随想录训练第二十一天|LeetCode93.复原IP地址、LeetCode78.子集、LeetCode90.子集II

君君超有趣

已于 2024-07-16 02:22:27 修改

阅读量544

点赞数 27

分类专栏：代码随想录文章标签：算法 java 数据结构 leetcode 剪枝

于 2024-07-16 01:41:25 首次发布

本文链接：https://blog.csdn.net/weixin_60364343/article/details/140453118

版权

代码随想录专栏收录该内容

39 篇文章 0 订阅

订阅专栏

文章目录

93.复原IP地址
- 思路
- 回溯三部曲
78.子集
- 思路
- 回溯三部曲
90.子集II
- 思路

93.复原IP地址

有效 IP 地址 正好由四个整数（每个整数位于 0 到 255 之间组成，且不能含有前导 0），整数之间用 '.' 分隔。

例如："0.1.2.201" 和 "192.168.1.1" 是有效 IP 地址，但是 "0.011.255.245"、"192.168.1.312" 和 "192.168@1.1" 是无效 IP 地址。

给定一个只包含数字的字符串 s ，用以表示一个 IP 地址，返回所有可能的有效 IP 地址，这些地址可以通过在 s 中插入 '.' 来形成。你不能重新排序或删除 s 中的任何数字。你可以按任何顺序返回答案。

示例 1：

输入：s = "25525511135"
输出：["255.255.11.135","255.255.111.35"]

示例 2：

输入：s = "0000"
输出：["0.0.0.0"]

示例 3：

输入：s = "101023"
输出：["1.0.10.23","1.0.102.3","10.1.0.23","10.10.2.3","101.0.2.3"]

提示：

1 <= s.length <= 20
s 仅由数字组成

思路

做这道题目之前，最好先把131.分割回文串 (opens new window)这个做了。

这道题目相信大家刚看的时候，应该会一脸茫然。

其实只要意识到这是切割问题，切割问题就可以使用回溯搜索法把所有可能性搜出来，和刚做过的131.分割回文串 (opens new window)就十分类似了。

切割问题可以抽象为树型结构，如图：

93.复原IP地址

回溯三部曲

递归参数

在131.分割回文串 (opens new window)中我们就提到切割问题类似组合问题。

startIndex一定是需要的，因为不能重复分割，记录下一层递归分割的起始位置。

本题我们还需要一个变量number，记录当前的段数。

所以代码如下：

//结果集
private List<String> res = new ArrayList<>();
//定义一个path用来记录遍历路径
private StringBuilder path = new StringBuilder();
/**
 * @Description
 * @Param s 字符串
 * @Param startIndex 分割起始位置
 * @Param number 当前段数
 * @Return
 * @Author 君君
 * @Date 2024/7/15 17:22
 */
private void travel(String s, int startIndex, int number)

递归终止条件

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

number表示当前的段数，number为4说明字符串分成了4段了。

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

但是如果起始下标已经等于了字符串的长度,或者当前段已经为四段,但是结果不符合要求,则返回上一层.

代码如下：

    //如果段数为四段,则说明已经分割结束,有四段字符串
    if (startIndex == s.length() && number == 4) {
        //这是需要判断第四段子串是否合法
        //如果合法的话,将当前段加入结果
        res.add(path.toString());
        return;
    }// 如果start等于s的长度但是ip段的数量不为4，或者ip段的数量为4但是start小于s的长度，则直接返回
    if (startIndex == s.length() || number == 4) {
        return;
    }

单层搜索的逻辑

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

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

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

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

93.复原IP地址

然后就是递归和回溯的过程：

递归调用时，下一层递归的startIndex要从i+1开始（因为是在新的字符串上边进行添加），同时记录段数number要 +1。

回溯的时候，就将刚刚加入的分隔符. 删掉就可以了，number也要-1,在这里要加上当前段数的判断,如果为。

代码如下：

for (int i = startIndex; i < s.length(); i++) {
    //当前段已经有了三个数,则退出循环
    if (i - startIndex > 3) {
        break;
    }
    //如果当前段符合要求,则加入path,并进入下一次递归下一次递归
    if (isValid(s, startIndex, i)) {
        String temp = s.substring(startIndex, i + 1);
        path.append(temp);

        // 当stringBuilder里的网段数量小于3时，才会加点；如果等于3，说明已经有3段了，最后一段不需要再加点
        if (number < 3) {
            path.append('.');
        }
        number++;
        //进入下一层递归
        travel(s, i + 1, number);
        //回溯
        number--;
        // 当stringBuilder里的网段数量小于3时，才会加点；如果等于3，说明已经有3段了，最后一段不需要再加点
        if (number < 3) {
            path.delete(path.length() - 1 - (i - startIndex) - 1, path.length());
        } else {
            path.delete(path.length() - 1 - (i - startIndex), path.length());
        }
    } else {
        break;
    }
}

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

private List<String> res = new ArrayList<>();
//定义一个path用来记录遍历路径
private StringBuilder path = new StringBuilder();

/**
 * @Description
 * @Param s 字符串
 * @Param startIndex 分割起始位置
 * @Param number 当前段数
 * @Return
 * @Author 君君
 * @Date 2024/7/15 17:22
 */
private void travel(String s, int startIndex, int number) {
    //如果分割点有三个,则说明已经分割结束,有四段字符串
    if (startIndex == s.length() && number == 4) {
        //这是需要判断第四段子串是否合法
        //如果合法的话,将当前段加入结果
        res.add(path.toString());
        return;
    }// 如果start等于s的长度但是ip段的数量不为4，或者ip段的数量为4但是start小于s的长度，则直接返回
    if (startIndex == s.length() || number == 4) {
        return;
    }

    for (int i = startIndex; i < s.length(); i++) {
        //当前段已经有了三个数,则退出循环
        if (i - startIndex > 3) {
            break;
        }
        //如果当前段符合要求,则加入path,并进入下一次递归下一次递归
        if (isValid(s, startIndex, i)) {
            String temp = s.substring(startIndex, i + 1);
            path.append(temp);

            //添加一个"."
            if (number < 3) {
                path.append('.');
            }
            number++;
            //进入下一层递归
            travel(s, i + 1, number);
            //回溯
            number--;
            if (number < 3) {
                path.delete(path.length() - 1 - (i - startIndex) - 1, path.length());
            } else {
                path.delete(path.length() - 1 - (i - startIndex), path.length());
            }
        } else {
            break;
        }
    }
}

/**
 * @Description 1. 以0开头的数字不合法
 * 2. 如果有非正整数的字符不合法
 * 3. 如果大于255则不合法
 * @Param s 字符串
 * @Param startIndex 判断起始位置
 * @Param endIndex 判断结束位置
 * @Return {@link boolean}
 * @Author 君君
 * @Date 2024/7/15 18:46
 */
private boolean isValid(String s, int startIndex, int endIndex) {
    if (startIndex > endIndex) {
        return false;
    }
    //开头为0的字段不合法
    if (s.charAt(startIndex) == '0' && endIndex - startIndex > 0) {
        return false;
    }
    //记录当前段的大小
    int sum = 0;
    for (int i = startIndex; i <= endIndex; i++) {
        //遍历单个字符
        if (s.charAt(i) > '9' || s.charAt(i) < '0') {
            return false;
        }
        sum = sum * 10 + s.charAt(i) - '0';
        if (sum > 255) {
            return false;
        }
    }
    return true;
}

时间复杂度: O(3^4)，IP地址最多包含4个数字，每个数字最多有3种可能的分割方式，则搜索树的最大深度为4，每个节点最多有3个子节点。
空间复杂度: O(n)

78.子集

给你一个整数数组 nums ，数组中的元素 互不相同 。返回该数组所有可能的子集（幂集）。

解集不能包含重复的子集。你可以按 任意顺序 返回解集。

示例 1：

输入：nums = [1,2,3]
输出：[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]

示例 2：

输入：nums = [0]
输出：[[],[0]]

提示：

1 <= nums.length <= 10
-10 <= nums[i] <= 10
nums 中的所有元素 互不相同

思路

求子集问题和77.组合 (opens new window)和131.分割回文串 (opens new window)又不一样了。

如果把子集问题、组合问题、分割问题都抽象为一棵树的话，那么组合问题和分割问题都是收集树的叶子节点，而子集问题是找树的所有节点！

其实子集也是一种组合问题，因为它的集合是无序的，子集{1,2} 和子集{2,1}是一样的。

那么既然是无序，取过的元素不会重复取，写回溯算法的时候，for就要从startIndex开始，而不是从0开始！

有同学问了，什么时候for可以从0开始呢？

求排列问题的时候，就要从0开始，因为集合是有序的，{1, 2} 和{2, 1}是两个集合，排列问题我们后续的文章就会讲到的。

以示例中nums = [1,2,3]为例把求子集抽象为树型结构，如下：

78.子集

从图中红线部分，可以看出遍历这个树的时候，把所有节点都记录下来，就是要求的子集集合。

回溯三部曲

递归函数参数

全局变量数组path为子集收集元素，二维数组result存放子集组合。（也可以放到递归函数参数里）

递归函数参数在上面讲到了，需要startIndex。

代码如下：

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

private void travel(int[] nums, int startIndex){
    
}

递归终止条件

从图中可以看出：

78.子集

剩余集合为空的时候，就是叶子节点。

那么什么时候剩余集合为空呢？

就是startIndex已经大于数组的长度了，就终止了，因为没有元素可取了，代码如下:

//长度已经遍历结束
    if (startIndex >= nums.length) {
        return;
    }

其实可以不需要加终止条件，因为startIndex >= nums.size()，本层for循环本来也结束了。

单层搜索逻辑

求取子集问题，不需要任何剪枝！因为子集就是要遍历整棵树。

那么单层递归逻辑代码如下：

    for (int i = startIndex; i < nums.length; i++) {
        //在此处进行丰富子集
        path.add(nums[i]);
        //进入下一层递归
        travel(nums, i + 1);
        path.remove(path.size() - 1);
    }

整体代码

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

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

private void travel(int[] nums, int startIndex) {
    //在此处收集子集
    res.add(new ArrayList<>(path));
    //长度已经遍历结束
    if (startIndex >= nums.length) {
        return;
    }
    for (int i = startIndex; i < nums.length; i++) {
        //在此处进行丰富子集
        path.add(nums[i]);
        //进入下一层递归
        travel(nums, i + 1);
        path.remove(path.size() - 1);
    }
}

时间复杂度: O(n * 2^n)
空间复杂度: O(n)

在注释中，可以发现可以不写终止条件，因为本来我们就要遍历整棵树。

有的同学可能担心不写终止条件会不会无限递归？

并不会，因为每次递归的下一层就是从i+1开始的。

90.子集II

给你一个整数数组 nums ，其中可能包含重复元素，请你返回该数组所有可能的子集（幂集）。

解集不能包含重复的子集。返回的解集中，子集可以按 任意顺序 排列。

示例 1：

输入：nums = [1,2,2]
输出：[[],[1],[1,2],[1,2,2],[2],[2,2]]

示例 2：

输入：nums = [0]
输出：[[],[0]]

提示：

1 <= nums.length <= 10
-10 <= nums[i] <= 10

思路

做本题之前一定要先做78.子集 (opens new window)。

这道题目和78.子集 (opens new window)区别就是集合里有重复元素了，而且求取的子集要去重。

那么关于回溯算法中的去重问题，在40.组合总和II (opens new window)中已经详细讲解过了，和本题是一个套路。

剧透一下，后期要讲解的排列问题里去重也是这个套路，所以理解“树层去重”和“树枝去重”非常重要。

用示例中的[1, 2, 2] 来举例，如图所示：

（注意去重需要先对集合排序）

90.子集II

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

本题就是其实就是回溯算法：求子集问题！ (opens new window)的基础上加上了去重，去重我们在回溯算法：求组合总和（三） (opens new window)也讲过了，所以我就直接给出代码了：

public List<List<Integer>> subsetsWithDup(int[] nums) {
    Arrays.sort(nums);
    used = new boolean[nums.length];
    Arrays.fill(used,false);
    travel(nums,0);
    return res;
}

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

private boolean[] used;
private void travel(int[] nums,int startIndex)
{
    //在开始先加入将子集入结果集
    res.add(new ArrayList<>(path));
    if(startIndex>= nums.length)
    {
        return;
    }
    for (int i = startIndex; i < nums.length; i++) {
        //如果是同层遇到同值元素则跳过本次循环
        if(i>0&&nums[i]==nums[i-1]&&!used[i-1])
        {
            continue;
        }
        used[i] = true;
        path.add(nums[i] );
        //递归
        travel(nums,i+1);
        path.remove(path.size()-1);
        used[i] = false;
    }
}

时间复杂度: O(n * 2^n)
空间复杂度: O(n)

君君超有趣

关注

27
点赞
踩
25

收藏

觉得还不错? 一键收藏
0
评论
代码随想录训练第二十一天|LeetCode93.复原IP地址、LeetCode78.子集、LeetCode90.子集II

递归调用时，下一层递归的startIndex要从i+1开始（因为是在新的字符串上边进行添加），同时记录段数number要 +1。求排列问题的时候，就要从0开始，因为集合是有序的，{1, 2} 和{2, 1}是两个集合，排列问题我们后续的文章就会讲到的。从图中可以看出，同一树层上重复取2 就要过滤掉，同一树枝上就可以重复取2，因为同一树枝上元素的集合才是唯一子集！其实子集也是一种组合问题，因为它的集合是无序的，子集{1,2} 和子集{2,1}是一样的。区别就是集合里有重复元素了，而且求取的子集要去重。
复制链接

扫一扫

专栏目录