LeetCode - Permutations II

最新推荐文章于 2019-03-26 13:08:28 发布

MingjaLee

最新推荐文章于 2019-03-26 13:08:28 发布

阅读量466

点赞数

分类专栏： ACM LeetCode 文章标签： leetcode backtrack c++

本文链接：https://blog.csdn.net/lijiang1991/article/details/51260154

版权

ACM 同时被 2 个专栏收录

13 篇文章 0 订阅

订阅专栏

LeetCode

6 篇文章 0 订阅

订阅专栏

题目链接：https://leetcode.com/problems/permutations-ii/

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

For example,
[1,1,2] have the following unique permutations:
[1,1,2], [1,2,1], and [2,1,1].

题目大意，列出所有非重复的排列组合。一般采用回溯（backtracking）递归求解，对于递归过程中标记已访问的数据有两种方法，一是开辟数组vis[]，二是利用原数据数组，显然第二种效率高一些，具体耗时为一（32ms），二（28ms）。第一种方法比较容易想到，一开始是我也只想到这一种，所以写博客记录一下思维的差距。

I 开辟vis[]数组

class Solution {
public:
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        vector<vector<int> > ret;
        if(nums.empty())
            return ret;
        sort(nums.begin(), nums.end());
        vector<bool> vis(nums.size(), false);
        vector<int> tmp;
        dfs(ret, nums, tmp, vis);
        return ret;
    }

    void dfs(vector<vector<int> > &ret, vector<int> &nums, vector<int> &tmp, vector<bool> &vis)
    {
        if(tmp.size() == nums.size()) {
            ret.push_back(tmp);
            return;
        }
        else {
            for(int i = 0; i < nums.size(); i++) {
                if(vis[i] == true)
                    continue;
                if(i > 0 && vis[i - 1] == false && vis[i] == false && nums[i] == nums[i - 1])
                    continue;
                tmp.push_back(nums[i]);
                vis[i] = true;
                dfs(ret, nums, tmp, vis);
                vis[i] = false;
                tmp.pop_back();
            }
        }
    }
};

II. 利用原数据数组标记

class Solution {
public:
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        vector<vector<int> > ret;
        if(nums.empty())
            return ret;
        sort(nums.begin(), nums.end()); //this is not necessary, but can raise efficiency
        dfs(ret, nums, 0);
        return ret;
    }

    //nums[0, begin) is visted, nums[begin, end] is not
    void dfs(vector<vector<int> > &ret, vector<int> &nums, int begin)
    {
        if(begin == nums.size() - 1) {
            ret.push_back(nums);
            return;
        }
        else {
            for(int i = begin; i < nums.size(); i++) {
                bool flag = false; // exit nums[j] == nums[i], where j range of [begin, i) ?
                for(int j = begin; j < i; j++) {
                    if (nums[i] == nums[j]) {
                        flag = true;
                        break;
                    }
                }
                if(flag == true)
                    continue;
                swap(nums[i], nums[begin]);
                dfs(ret, nums, begin + 1);
                swap(nums[i], nums[begin]);
            }
        }
    }
};