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

酱油黑龙發表於2024-10-02
演算法

題目1 491. 非遞減子序列

給你一個整數陣列 nums ，找出並返回所有該陣列中不同的遞增子序列，遞增子序列中 至少有兩個元素 。你可以按 任意順序 返回答案。

陣列中可能含有重複元素，如出現兩個整數相等，也可以視作遞增序列的一種特殊情況。

示例 1：

輸入：nums = [4,6,7,7]
輸出：[[4,6],[4,6,7],[4,6,7,7],[4,7],[4,7,7],[6,7],[6,7,7],[7,7]]

示例 2：

輸入：nums = [4,4,3,2,1]
輸出：[[4,4]]

提示：

  • 1 <= nums.length <= 15
  • -100 <= nums[i] <= 100

程式碼

class Solution {
public:
    vector<vector<int>> result;
    vector<int> path;
    void pathtracking(vector<int>& nums, int curIndex)
    {
        set<int> numset;
        if(path.size() >= 2)
        {
            result.push_back(path);
        }
        for(int i = curIndex; i < nums.size(); i++)
        {
            if((!path.empty() && path.back() > nums[i] )|| 
                numset.find(nums[i]) != numset.end())
                continue;
            numset.insert(nums[i]);
            path.push_back(nums[i]);
            pathtracking(nums, i + 1);
            path.pop_back();
        }
    }
    vector<vector<int>> findSubsequences(vector<int>& nums) {
        if(nums.size() == 1)
            return vector<vector<int>>();
        pathtracking(nums, 0);
        return result;
    }
};

題目2 46. 全排列

給定一個不含重複數字的陣列 nums ，返回其 所有可能的全排列 。你可以 按任意順序 返回答案。

示例 1：

輸入：nums = [1,2,3]
輸出：[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]

示例 2：

輸入：nums = [0,1]
輸出：[[0,1],[1,0]]

示例 3：

輸入：nums = [1]
輸出：[[1]]

提示：

  • 1 <= nums.length <= 6
  • -10 <= nums[i] <= 10
  • nums 中的所有整數 互不相同

程式碼

class Solution {
public:
    bool hash[6] = {false};
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(vector<int>& nums)
    {
        if(path.size() == nums.size())
        {
            result.push_back(path);
            return;
        }
        for(int i = 0; i < nums.size(); i++)
        {
            if(hash[i])
                continue;
            path.push_back(nums[i]);
            hash[i] = true;
            backtracking(nums);
            path.pop_back();
            hash[i] = false;
        }
    }
    vector<vector<int>> permute(vector<int>& nums) {
        backtracking(nums);
        return result;
    }
};

題目3 47. 全排列 II

給定一個可包含重複數字的序列 nums ，按任意順序 返回所有不重複的全排列。

示例 1：

輸入：nums = [1,1,2]
輸出：
[[1,1,2],
 [1,2,1],
 [2,1,1]]

示例 2：

輸入：nums = [1,2,3]
輸出：[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]

提示：

  • 1 <= nums.length <= 8
  • -10 <= nums[i] <= 10

程式碼

class Solution {
public:
    vector<vector<int>> result;
    vector<int> path;
    bool hash[8] = {false};
    void backtracking(vector<int>& nums)
    {
        if(path.size() == nums.size())
        {
            result.push_back(path);
            return;
        }
        unordered_set<int> uset;
        for(int i = 0; i < nums.size(); i++)
        {
            if(hash[i] || uset.find(nums[i]) != uset.end())
                continue;
            uset.insert(nums[i]);
            path.push_back(nums[i]);
            hash[i] = true;
            backtracking(nums);
            path.pop_back();
            hash[i] = false; 
        }
    }
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        backtracking(nums);
        return move(result);
    }
};

相關文章