LeetCode-47-Permutations II(DFS/剪枝)-Medium

原创于 2016-01-10 14:37:01 发布 · 849 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

68 篇文章

订阅专栏

本文探讨了在46题的基础上增加元素可能重复的情况，并通过深度优先搜索(DFS)实现解决方案，注意剪枝操作以避免重复解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意理解：

同46，但vector中的元素可能存在重复值；

题目分析：

同46一样通过DFS求解，但注意剪枝；

解题代码：

class Solution {
private:
    vector<vector<int>> ans;
    bool findValue(vector<int> &v, int endIndex, int val){
        for(int i=0; i<endIndex; i++){
            if(v[i]==val){
                return true;
            }
        }    
        
        return false;
    }
    
    void dfs(vector<int> &preAns, vector<int> &subNums){
        if(subNums.size()==0){
            ans.push_back(preAns);
            return;
        }
        
        int size=subNums.size();
        for(int i=0; i<size; i++){
            int n=subNums[i];
            if(findValue(subNums, i, n)){
                continue;
            }
            
            vector<int> tPreAns(preAns);
            tPreAns.push_back(n);
            
            vector<int> tSubNums(subNums);
            tSubNums.erase(find(tSubNums.begin(), tSubNums.end(),n));
            dfs(tPreAns, tSubNums);
        }
    }
    
public:
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        vector<int> pre;
        dfs(pre, nums);
        
        return ans;
    }
};