全排列 II

问题描述

给定一个可包含重复数字的序列,返回所有不重复的全排列。

示例:

输入: [1,1,2]
输出:
[
  [1,1,2],
  [1,2,1],
  [2,1,1]
]

思路

  • 回溯算法,得到所有的排序情况ret;
  • 对ret进行排序,使用ret.erase(unique(ret.begin(), ret.end()), ret.end())对ret进行去重

题解

class Solution {
public:
    vector<vector<int>> ret;
    vector<int> tmp;
    int n;
    bool visited[100] = {false};

    vector<vector<int>> permuteUnique(vector<int> &nums) {
        n = nums.size();
        if (n == 0) return ret;
        dfs(nums);
        sort(ret.begin(), ret.end());
        ret.erase(unique(ret.begin(), ret.end()), ret.end());
        return ret;
    }

    void dfs(vector<int> nums) {
        if (tmp.size() == n) {
            ret.push_back(tmp);
            return;
        }
        for (int i = 0; i < n; i++) {
            if (visited[i] == false) {
                visited[i] = true;
                tmp.push_back(nums[i]);
                dfs(nums);
                tmp.pop_back();
                visited[i] = false;
            }
        }
    }
};

标签: none

添加新评论