Day28 回溯第四天
LeetCode 93. 复原 IP 地址
这道题目的难点同样是题意,有了回文串的前车之鉴,这道题的分割倒是有迹可循,但这道题判断是否合理同样也有很多细节。
class Solution {
public:vector<string> res;bool isValid(const string&s,int start,int end){if(start>end) return false;if(s[start]=='0' && start!=end) return false;int num=0;for(int i=start;i<=end;i++){if(s[i]>'9' || s[i]<'0') return false;num=num*10+(s[i]-'0');if(num>255) return false;}return true;}void backtrack(string& s,int startIndex,int pointSum){if(pointSum==3){if(isValid(s,startIndex,s.size()-1))res.push_back(s);return;}for(int i=startIndex;i<s.size();i++){if(isValid(s,startIndex,i)){s.insert(s.begin()+i+1,'.');pointSum++;backtrack(s,i+2,pointSum);pointSum--;s.erase(s.begin()+i+1);}else break;}}vector<string> restoreIpAddresses(string s) {if(s.size()<4 || s.size()>12) return res;backtrack(s,0,0);return res;}
};
LeetCode 78.子集
子集问题在代码方面已经是容易的回溯了。
但子集问题和之前的问题的最大区别在于:子集问题是每一层的遍历结果都是结果中元素,所以每次递归都要把当前的递归结果放到结果集中,而不是终止时才添加。
class Solution {
public:vector<int> path;vector<vector<int>> res;void backtrack(vector<int>& nums,int startIndex){res.push_back(path);if(startIndex>=nums.size()) return;for(int i=startIndex;i<nums.size();i++){path.push_back(nums[i]);backtrack(nums,i+1);path.pop_back();}}vector<vector<int>> subsets(vector<int>& nums) {backtrack(nums,0);return res;}
};
LeetCode 90.子集II
带重复元素的子集问题和前面一道组合总和时的思路完全相同,利用used数组去重,然后不断向前遍历防止重复,相当于两道题的思路的组合了。
class Solution {
public:vector<int> path;vector<vector<int>> res;void backtrack(vector<int>& nums,int startIndex,vector<bool>& used){res.push_back(path);for(int i=startIndex;i<nums.size();i++){if(i>0 && nums[i]==nums[i-1] && used[i-1]==false)continue;path.push_back(nums[i]);used[i]=true;backtrack(nums,i+1,used);used[i]=false;path.pop_back();}}vector<vector<int>> subsetsWithDup(vector<int>& nums) {vector<bool> used(nums.size(),false);sort(nums.begin(),nums.end());backtrack(nums,0,used);return res;}
};
今天又学习了子集问题,加油!