算法训练营 第二十九天

算法训练营 第二十九天

回溯算法的第五天

491.递增子序列

这道题的难点是如何处理重复的path；方法用的之前提到的树层去重。这个方法很有用，需要牢记。

class Solution {
public:
    void backtracking(const vector<int>& nums,int startIndex) {
        if (path.size() > 1) {
            result.push_back(path);
        }
        unordered_set<int> uset;
        for (int i = startIndex; i < nums.size(); i++) {
            
            if (!path.empty() && path.back() > nums[i]) {
                continue;
            }
            if (uset.find(nums[i]) != uset.end()) {
                continue;
            }
            uset.insert(nums[i]);
            path.push_back(nums[i]);
            backtracking(nums, i + 1);
            path.pop_back();
            
        }
    }
    vector<vector<int>> findSubsequences(vector<int>& nums) {
        backtracking(nums, 0);
        return result;
    }
private:
    vector<int> path;
    vector<vector<int>> result;
};

46.全排列

这道题用used数组记录一下nums的使用情况就行。

class Solution {
public:
    
    vector<vector<int>> permute(vector<int>& nums) {
        used = vector<int>(nums.size(), 0);
        backtracking(nums);
        return res;
    }
private:
    vector<int> used;
    vector<int> path;
    vector<vector<int>> res;
    void backtracking(const vector<int>& nums) {
        if (path.size() == nums.size()) {
            res.push_back(path);
        }
        for (int i = 0; i < nums.size(); i++) {
            if (used[i] == 1) {
                continue;
            }
            used[i] = 1;
            path.push_back(nums[i]);
            backtracking(nums);
            path.pop_back();
            used[i] = 0;
        }
    }
};

47.全排列 II

这道题需要树层去重，可以将nums排序之后再用used处理（能对nums排序处理的题就用used，不能排序的话就用uset）；注意这道题用树枝去重也行，树枝去重了树层就不用去重了。

class Solution {
private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking (vector<int>& nums, vector<bool>& used) {
        // 此时说明找到了一组
        if (path.size() == nums.size()) {
            result.push_back(path);
            return;
        }
        for (int i = 0; i < nums.size(); i++) {
            // used[i - 1] == true，说明同一树枝nums[i - 1]使用过
            // used[i - 1] == false，说明同一树层nums[i - 1]使用过
            // 如果同一树层nums[i - 1]使用过则直接跳过
            if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {
                continue;
            }
            if (used[i] == false) {
                used[i] = true;
                path.push_back(nums[i]);
                backtracking(nums, used);
                path.pop_back();
                used[i] = false;
            }
        }
    }
public:
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        result.clear();
        path.clear();
        sort(nums.begin(), nums.end()); // 排序
        vector<bool> used(nums.size(), false);
        backtracking(nums, used);
        return result;
    }
};