C++算法:組合總和------遞歸回溯

leetcode相關C++算法解答: https://github.com/Nereus-Minos/C_plus_plus-leetcode

題目:

給定一個無重複元素的數組 candidates 和一個目標數 target ,找出 candidates 中所有可以使數字和爲 target 的組合。candidates 中的數字可以無限制重複被選取。
說明: 所有數字(包括 target)都是正整數。 解集不能包含重複的組合。

示例 1:

輸入: candidates = [2,3,6,7], target = 7, 所求解集爲: [ [7], [2,2,3] ]

示例 2:

輸入: candidates = [2,3,5], target = 8, 所求解集爲: [[2,2,2,2],[2,3,3],[3,5] ]

思路:

使用遞歸回溯的方法求出所有接
關於去重方法:1.使用map(太複雜,且效果不好)2.在遞歸的時候將start也傳入

代碼:


#if 0
//思路:爲了避免重複,使用map去重
class Solution {
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        
        vector<vector<int>> ret;
        vector<int> temp;
        map<vector<int>, int> ret_temp; //爲了去重
        
        search_combine(candidates, temp, ret_temp, target, 0);
        
        for(map<vector<int>, int>::iterator iter = ret_temp.begin(); iter != ret_temp.end(); iter++) //iter != ret_temp.end()沒有重載<比較符,所以只能用!=
            ret.push_back((*iter).first);
        
        return ret;
    }
    
private:
    void search_combine(vector<int>& candidates, vector<int> temp, map<vector<int>, int>& ret_temp, int& target, int count){
        if(count == target)
        {
            sort(temp.begin(), temp.end());
            if(ret_temp.find(temp) == ret_temp.end())
                ret_temp[temp] = 1;
            
            return ;
        }
        else if(count > target)
            return ;
        else
        {
            for(int i = 0; i < candidates.size(); i++)
            {
                temp.push_back(candidates[i]);
                count += candidates[i];
                search_combine(candidates, temp, ret_temp, target, count);
                temp.erase(temp.end()-1);
                count -= candidates[i];
            }
        }
    }
};
#endif

#if 1
//思路:爲了避免重複,需要在遞歸的時候將start也傳入
class Solution {
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        
        vector<vector<int>> ret;
        vector<int> temp;
        
        sort(candidates.begin(), candidates.end());
        
        search_combine(candidates, temp, ret, target, 0, 0);
        
        return ret;
    }
    
private:
    void search_combine(vector<int>& candidates, vector<int> temp, vector<vector<int>>& ret, int& target, int count, int start){
        if(count == target)
        {
            ret.push_back(temp);
            
            return ;
        }
        else if(count > target)
            return ;
        else
        {
            for(int i = start; i < candidates.size(); i++)
            {
                temp.push_back(candidates[i]);
                count += candidates[i];
                search_combine(candidates, temp, ret, target, count, i);
                temp.erase(temp.end()-1);
                count -= candidates[i];
            }
        }
    }
};
#endif
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章