LeetCode 47 全排列II(有重复元素 dfs)

LeetCode 47 全排列II(有重复元素 dfs)
对于有重复数字的全排列,分析重复情况的主要来源:相同的位置选择了不同但数值相同的数字。
对于重复的数字,人为控制放入相同数字的数量(枚举从1~N),只要保证选择i个连续相同数字后,紧跟的数字不相同(不能构成连续的i+1个相同数字)即可保证排列唯一性。
tricks:用map存储每个数字的个数。枚举for循环结束后,统一pop出N个相同数字维护当前排列。

附上代码:

class Solution {
    map<int,int> vis;
public:
    void dfs(vector<int>& nums, vector<vector<int>>& out, vector<int>& curr){
        if(curr.size() >= nums.size()) {
            out.push_back(curr); return;
        }
        for(map<int,int>::iterator it = vis.begin(); it != vis.end(); it++){
            if(curr.size() > 0 && curr.back() == it->first) continue;
            int temp = it->second;
            while(it->second > 0) {
                vis[it->first]--;
                curr.push_back(it->first);
                dfs(nums, out, curr);
            }
            vis[it->first] = temp;
            while(temp > 0){
                curr.pop_back();
                temp--;
            }
        }
    }

    vector<vector<int>> permuteUnique(vector<int>& nums) {
        vector<vector<int>> out;
        vector<int> current;
        sort(nums.begin(), nums.end());
        for(auto n:nums){
            if(vis.count(n) == 0) vis[n] = 1;
            else vis[n]++;
        }
        dfs(nums, out, current);
        return out;
    }
};
上一篇:2021年11.13周总结


下一篇:Codeforces Round #749 (Div.1+Div.2)A-D题解