代码随想录算法训练营二十一天|回溯part03
LeetCode 93 复原IP地址
题目链接:93. 复原 IP 地址 - 力扣(LeetCode)
有效 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"]
这个问题和分割回文串一样,都是切割问题,抽象为树型结构如图所示:
回溯函数参数:字符串s,起始位置startIndex
终止条件:字符串已经被分为四段
单层搜索的逻辑:和分割回文串相似(具体见我的上一篇博客)
代码如下:
class Solution {List<String> res=new ArrayList<>();List<String> list=new ArrayList<>();public List<String> restoreIpAddresses(String s) {if(s.length()<4||s.length()>12)return res;backtracking(s,0);return res;}public void backtracking(String s,int startIndex){if(list.size()==4&&startIndex>=s.length()){StringBuffer sb=new StringBuffer();for(int i=0;i<list.size()-1;i++){sb.append(list.get(i));sb.append(".");}sb.append(list.get(list.size()-1));res.add(sb.toString());return;}for(int i=startIndex;i<s.length();i++){if(s.length()-startIndex>(4-list.size())*3||s.length()-startIndex<(4-list.size()))break;//剩余字符太多或太少都无法构成合法IPString s1=s.substring(startIndex,i+1);long num=Long.parseLong(s1);if(num>=0&&num<=255&&(s1.charAt(0)-'0'!=0||s1.length()==1)){//判断数字是否合法list.add(s1);backtracking(s,i+1);list.remove(list.size()-1);}}}
}
LeetCode 78 子集
题目链接:78. 子集 - 力扣(LeetCode)
给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。
示例 1:
输入:nums = [1,2,3]
输出:[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]
示例 2:输入:nums = [0]
输出:[[],[0]]
本题抽象为树型结构:
直接套回溯模板就行 :
class Solution {List<List<Integer>> res=new ArrayList<List<Integer>>();List<Integer> list=new ArrayList<>();public List<List<Integer>> subsets(int[] nums) {res.add(new ArrayList<>(list));backtracking(nums,0);return res;}public void backtracking(int[] arr,int startIndex){for(int i=startIndex;i<arr.length;i++){list.add(arr[i]);res.add(new ArrayList<>(list));backtracking(arr,i+1);list.remove(list.size()-1);}}
}
LeetCode 90 子集II
题目链接:90. 子集 II - 力扣(LeetCode)
给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的 子集(幂集)。
解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。
示例 1:
输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]
示例 2:输入:nums = [0]
输出:[[],[0]]
这个题相当于是78.子集和40.组合总和II两道题的结合,一方面要递归求子集,一方面还要去重。
直接上代码,去重的思路上一篇博客已经说过了:
class Solution {List<List<Integer>> res=new ArrayList<List<Integer>>();List<Integer> list=new ArrayList<>();public List<List<Integer>> subsetsWithDup(int[] nums) {res.add(new ArrayList<>(list));Arrays.sort(nums);backtracking(nums,0);return res;}public void backtracking(int[] arr,int startIndex){for(int i=startIndex;i<arr.length;i++){if(i>startIndex&&arr[i]==arr[i-1])continue;list.add(arr[i]);res.add(new ArrayList<>(list));backtracking(arr,i+1);list.remove(list.size()-1);}}
}
在之前回溯的基础之上,今天的三道题都很容易解出来。