leetcode—5/6

1.組合總和

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

void backtrade(vector<int>& candidates, int start, int target, vector<int>& temp, vector< vector<int> >& res)
    {
        if(target < 0)
        {
            return;
        }
        if(target == 0)
        {
            res.push_back(temp);
            return;
        }
        for(int i = start; i < candidates.size(); i++)
        {
            temp.push_back(candidates[i]);
            backtrade(candidates, i, target - candidates[i], temp, res);
            temp.pop_back();
        }
    }
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        vector< vector<int> > res;  //存放最終的解空間
        vector<int> temp;   //存放臨時的解空間
        backtrade(candidates, 0, target, temp, res);
        return res;
    }

2.組合總和II

給定一個無重複元素的數組candidates和一個目標數target,找出candidates中所有可以使數字和爲target的組合。
candidates中的每個數字在每個組合中只能使用一次。
所有數字(包括target)都是正整數。
解集不能包含重複的組合。

思路:對數組先進行排序防止重複

void backtrade(vector<int>& candidates, int start, int target, vector<int>& temp, vector< vector<int> >& res)
    {
        if(target < 0)
        {
            return;
        }
        if(target == 0)
        {
            res.push_back(temp);
            return;
        }
        for(int i = start; i < candidates.size(); i++)
        {
            if(i > start && candidates[i] == candidates[i - 1])
            {
                continue;
            }
            temp.push_back(candidates[i]);
            backtrade(candidates, i + 1, target - candidates[i], temp, res);
            temp.pop_back();
        }
    }
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        vector< vector<int> > res;  //存放全局解空間
        vector<int> temp;   //存放每次回溯的臨時解空間
        sort(candidates.begin(), candidates.end());
        backtrade(candidates, 0, target, temp, res);
        return res;
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章