AcWing 51. 数字排列

题目描述

输入一组数字(可能包含重复数字),输出其所有的排列方式。

样例

输入:[1,2,3]

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

问题分析 

此题大体上参照 leetcode 46题的思想,但是不同的是此题给的数组中有重复的元素,所以要想办法去重。涉及到去重的问题首先想到排序,于是在DFS之前,先对给定数组进行排序。排序过后,数组中重复的元素会挨着,所以在for循环中,如果当前的数和它前一个数相同并且前一个数没被用过,说明这是重复的,直接跳到下一次循环。以上就是和46题的区别以及改动,其实改动不大,除了先排序之外,只有一个地方改动了。

代码实现

class Solution {
public:
    vector<vector<int>> permutation(vector<int>& nums) {
        vector<vector<int>> ans;
        vector<int> out;
        vector<bool> point(nums.size(), false);
        sort(nums.begin(), nums.end());
        DFS(nums, 0, point, out, ans);
        return ans;
    }
    
    void DFS(vector<int>& nums, int level, vector<bool>& point, vector<int>& out, vector<vector<int>>& ans){
        if(level==nums.size()){
            ans.push_back(out);
            return;
        }
        for(int i=0;i<nums.size();i++){
            if(point[i]==true || (i > 0 && nums[i] == nums[i - 1] && point[i - 1] == false))
                continue;
            out.push_back(nums[i]);
            point[i] = true;
            DFS(nums, level+1, point, out, ans);
            out.pop_back();
            point[i] = false;
        }
    }
};


 

猜你喜欢

转载自blog.csdn.net/mengyujia1234/article/details/90056422