程式碼隨想錄演算法 - 回溯3

酱油黑龙發表於2024-09-20
演算法

明天開始建模比賽，沒時間寫思路了

題目1 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 僅由數字組成

程式碼

class Solution {
public:
    string str;
    vector<string> path;
    vector<string> result;
    bool ispartAddr()
    {
        if(str[0] == '0' && str.size() > 1)
            return false;
        int num = atoi(str.c_str());
        if(num >= 0 && num <=255)
            return true;
        else
            return false;
    }
    void backtracking(string& s, int curIndex)
    {
        if(path.size() == 4 && curIndex == s.size())
        {
            str = "";
            for(int i = 0; i < path.size() - 1; i++)
                str += path[i] + ".";
            str += path[3];
            result.push_back(str);
            return;
        }
        for(int i = curIndex; i < s.size(); i++)
        {
            str = s.substr(curIndex, i - curIndex + 1);
            if(!ispartAddr())
                break;
            path.push_back(str);
            backtracking(s, i + 1);
            path.pop_back();
        }
    }
    vector<string> restoreIpAddresses(string s) {
        backtracking(s, 0);
        return result;
    }
};

題目2 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 中的所有元素 互不相同

程式碼

class Solution {
public:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(vector<int>& nums, int curIndex)
    {
        if(curIndex <= nums.size())
        {
            result.push_back(path);
        }
        for(int i = curIndex; i < nums.size(); i++)
        {
            path.push_back(nums[i]);
            backtracking(nums, i + 1);
            path.pop_back();
        }
    }
    vector<vector<int>> subsets(vector<int>& nums) {
        backtracking(nums, 0);
        return result;
    }
};

題目3 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

程式碼

class Solution {
public:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(vector<int>& nums, int startIndex)
    {
        if(startIndex <= nums.size())
        {
            result.push_back(path);
        }
        for(int i = startIndex; i < nums.size(); i++)
        {
            path.push_back(nums[i]);
            backtracking(nums, i + 1);
            path.pop_back();
            while(i + 1 < nums.size() && nums[i + 1] == nums[i])
                i++;
        }
    }
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        backtracking(nums, 0);
        return result;
    }
};

相關文章