代码随想录算法训练营第24天 | 93.复原IP地址、78.子集、90.子集Ⅱ
文章目录
93.复原IP地址
解题思路
这是一个切割问题,可以使用回溯搜索法。
- 确定递归函数的参数和返回值:首先需要 startIndex,用于记录下一层递归分割的起始位置,另外还需要一个变量 pointNum,用于记录添加 " . " 的数量。
- 递归的终止条件:因为IP地址格式为四段,所以本题将分割的段数作为终止条件。
- 单层递归的逻辑:在上一题判断回文子串中涉及到过如何截取子字符串,本题中主要需要判断这个子字符串是否合法,考虑以下三点:以0开头的数字不合法;含非正整数字符不合法;大于255不合法
代码实现
class Solution {
public:
vector<string> result;
void backtracking(string& s, int startIndex, int pointNum) {
if (pointNum == 3) {
if (isValid(s, startIndex, s.size() - 1)) {
result.push_back(s);
}
return;
}
for (int i = startIndex; i < s.size(); i++) {
if (isValid(s, startIndex, i)) {
s.insert(s.begin() + i + 1, '.');
pointNum++;
backtracking(s, i + 2, pointNum);
pointNum--;
s.erase(s.begin() + i + 1);
} else {
break;
}
}
}
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;
}
vector<string> restoreIpAddresses(string s) {
result.clear();
if (s.size() > 12)
return result;
backtracking(s, 0, 0);
return result;
}
};
题目总结
本题将分割字符串的点覆盖,并且增加了操作字符串添加 " . " 作为分隔符,并验证区间的合法性,比较综合。
78.子集
解题思路
子集问题也是一种组合问题,同样使用回溯算法解决。
- 递归函数参数和返回值:path 为子集收集元素,二维数组 result 用于存放子集组合,另外需要 startIndex
- 终止条件:剩余集合为空时,到达了叶子节点
- 单层递归:遍历整棵树
代码实现
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex) {
result.push_back(path);
if (startIndex >= nums.size()) {
return;
}
for (int i = startIndex; i < nums.size(); i++) {
path.push_back(nums[i]);
backtracking(nums, i + 1);
path.pop_back();
}
}
vector<vector<int>> subsets(vector<int>& nums) {
path.clear();
result.clear();
backtracking(nums, 0);
return result;
}
};
题目总结
这题比较简单,基本是直接套用回溯的模板结构。
90.子集Ⅱ
解题思路
本题和上题的区别就在于集合中有重复元素了,需要进行去重操作
代码实现
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex, vector<bool>& used) {
result.push_back(path);
if (startIndex >= nums.size()) {
return;
}
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;
backtracking(nums, i + 1, used);
used[i] = false;
path.pop_back();
}
}
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
path.clear();
result.clear();
vector<bool> used(nums.size(), false);
sort(nums.begin(), nums.end());
backtracking(nums, 0, used);
return result;
}
};
题目总结
若本题是求全排列,则每次要从0开始遍历,为了跳过已入递归栈元素,需要使用 used 数组。