【深度优先搜索DFS】Lintcode 16. 带重复元素的排列

Lintcode 16. 带重复元素的排列

题目描述:给出一个具有重复数字的列表,找出列表所有不同的排列。
在这里插入图片描述
DFS小知识点:
DFS递归语句的前后一定是对称的操作,前面的操作在递归之后要进行撤销。每次递归其实只是一种尝试。

此题需要考虑两种重复:

  1. 同一个数字不能取两次。
  2. [1,2’,2"] 和 [1,2",2’]这种重复需要去除。

这里借助一个bool型的visited数组用来记录每个数字之前是否被访问过,来进行去重。代码采用递归方式实现:

class Solution {
    
    
public:
    /*
     * @param :  A list of integers
     * @return: A list of unique permutations
     */
    vector<vector<int>> permuteUnique(vector<int> &nums) {
    
    
        vector<vector<int>> result;
        if (0 == nums.size()) {
    
    
            result.push_back(vector<int>());//当nums为空[]时,返回[]
            return result;
        }
        
        sort(nums.begin(), nums.end());//先进行排序便于后续去重
        vector<int> tmpPermute;
        vector<bool> visited(nums.size(), false);//标记某值是否已经被用过
        helper(nums, tmpPermute, visited, result);
        return result;
    }
    
    //1. 递归的定义
    void helper(vector<int> &nums, vector<int> &tmpPermute, vector<bool> &visited, vector<vector<int>> &result) {
    
    
        //3. 递归的出口
        if (tmpPermute.size() == nums.size()) {
    
    
            result.push_back(tmpPermute);
            return;
        }
        
        //2. 递归的拆解
        for (int i = 0; i < nums.size(); ++i) {
    
    
            //同一个数字不能重复取
            if (visited[i]) {
    
    
                continue;
            }
            
            // [1,2',2"] 和 [1,2",2]
            //去除上面这种重复
            if (0 != i && nums.at(i) == nums.at(i - 1) && !visited[i - 1]) {
    
    
                continue;
            }
            
            tmpPermute.push_back(nums.at(i));
            visited[i] = true;
            helper(nums, tmpPermute, visited, result);//每次递归都只是一次尝试
            tmpPermute.pop_back();//递归语句前后对称
            visited[i] = false;
        }
    }
};

猜你喜欢

转载自blog.csdn.net/phdongou/article/details/113776288