permutation and its variations

最新推荐文章于 2023-09-20 15:51:13 发布

Phantomape

最新推荐文章于 2023-09-20 15:51:13 发布

阅读量263

点赞数

分类专栏： Algorithm LeetCode

本文链接：https://blog.csdn.net/Phantomape/article/details/53571561

版权

Algorithm 同时被 2 个专栏收录

11 篇文章 0 订阅

订阅专栏

LeetCode

10 篇文章 0 订阅

订阅专栏

Permutations

If we change the argument $vector<int> nums$

class Solution {
public:
    vector<vector<int>> permute(vector<int>& nums) {
        vector<vector<int>> res;
        if(nums.empty())    return res;
        sort(nums.begin(), nums.end());
        DFS(res, nums, 0);
        return res;
    }

    void DFS(vector<vector<int>>& res, vector<int> nums, int start){
        if(start == nums.size() - 1){
            res.push_back(nums);
            return;
        }
        for(int i = start; i < nums.size(); i++){
            swap(nums[i], nums[start]);
            DFS(res, nums, start + 1);
        }
    }
};

If we change the argument $vector<int> nums$ into reference, we would need another swap:

swap(nums[i], nums[start]);
DFS(res, nums, start + 1);
swap(nums[i], nums[start]);

Permutations II

class Solution {
public:
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        vector<vector<int>> res;
        if(nums.empty())    return res;
        sort(nums.begin(), nums.end());
        DFS(res, nums, 0);
        return res;
    }

    void DFS(vector<vector<int>>& res, vector<int> nums, int start){
        if(start == nums.size() - 1){
            res.push_back(nums);
            return;
        }
        for(int i = start; i < nums.size(); i++){
            if(i != start && nums[i] == nums[start])
                continue;
            swap(nums[i], nums[start]);
            DFS(res, nums, start + 1);
        }
    }
};

If we change the argument $vector<int> nums$ into reference, we would need a few lines of modifications:

    void helper(vector<vector<int>>& res, vector<int>& nums, int pos) {

        if (pos == nums.size()) {
            res.push_back(nums);
        } else {
            for (int i = pos; i < nums.size(); ++i) {
                if (i > pos && nums[i] == nums[pos]) continue;
                swap(nums[pos], nums[i]);
                helper(res, nums, pos + 1);

            }
            // restore nums
            for (int i = nums.size() - 1; i > pos; --i) {
                swap(nums[pos], nums[i]);
            }
        }
    }