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;
}
};
今天又学习了子集问题,加油!