LeetCode 47. Permutation II

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zhaohaibo_/article/details/86481869

Permutations II - LeetCode

Given a collection of numbers that might contain duplicates, return all possible unique permutations.

Example:

Input: [1,1,2]
Output:
[
  [1,1,2],
  [1,2,1],
  [2,1,1]
]

STL:

class Solution {
public:
    vector<vector<int>>ans;
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        do {
            ans.push_back(nums);
        } while (next_permutation(nums.begin(), nums.end()));
        return ans;
    }
};

DFS:
先把所有数排序,如果遇到连续相同的两个数,就跳过

class Solution {
public:
    vector<vector<int>> ans;
    vector<int> path;
    vector<bool> vis;
    
    vector<vector<int>> permute(vector<int>& nums) {
        vis = vector<bool>(nums.size(), false);
        path = vector<int>(nums.size());
        dfs(nums, 0);
        return ans;
    }
    
    void dfs(vector<int>& nums, int step) {
        if (step == nums.size()) {
            ans.push_back(path);
            return ;
        }
        for (int i = 0; i < nums.size(); i ++ ) {
            if(!vis[i]){
                vis[i] = true, path[i] = nums[step];
                dfs(nums, step + 1);
                vis[i] = false;
            }
        }
    }
};

猜你喜欢

转载自blog.csdn.net/zhaohaibo_/article/details/86481869