leetcode第38题--Combination Sum

题目:

Given a set of candidate numbers (C) and a target number (T), find all unique combinations in C where the candidate numbers sums to T.

The same repeated number may be chosen from C unlimited number of times.

Note:

  • All numbers (including target) will be positive integers.
  • Elements in a combination (a1, a2, … , ak) must be in non-descending order. (ie, a1 ≤ a2 ≤ … ≤ ak).
  • The solution set must not contain duplicate combinations.

For example, given candidate set 2,3,6,7 and target 7
A solution set is: 
[7] 
[2, 2, 3] 

题目的意思是给定一个candidate集合,里面包含一些数字,给定一个目标T,在集合中找出可以组合成T的所有组合,集合中的数字可以重复利用。但组合后的结果不能重复。结果中按非降序排列。举例集合C为 2,3,6,7 目标位7,那么就有两种组合满足,一个是[2,2, 3] 一个是[7]

思路:

利用回溯法,深度优先。剪枝判断是相加的数大于目标时返回,或者相加之和为目标的时候记录在ans中并返回。如果相加小于目标,那就往下一层找,找到满足要返回的条件为止。

class Solution {

public:

int sum38(vector<int> vec) // 返回vec中所有值的和,用来和target判断用

{

    if (vec.size() == 0)

        return -1;

    int len = vec.size(), ans = 0;

    for (int i = 0; i < len; ++i)

    {

        ans += vec[i];

    }

    return ans;

}

void back38(vector<vector<int> > &ans, vector<int> &candidates, vector<int> &tmp, int target, int cur)

{

    for (int i = cur; i < candidates.size(); ++i) // i从传入的cur下标开始,cur初始为0,之后根据i改变,为了保证不会有重复解

    {

        int num = sum38(tmp); // 避免三个if语句都调用sum38,计算一次就好,否则超时

        if (num == target)

        {

            ans.push_back(tmp); // 如果找到了一组,就记录后返回

            return;

        }

        if (num < target) // 如果比目标还小,说明还要在加一个数

        {

            tmp.push_back(candidates[i]);

            back38(ans, candidates, tmp, target, i); // tmp加了一个数后再递归,注意此处传入的cur为i,这样为了保证不会有重复的解

                                                     // 因为candidates是排好序的,所以之后递归中要加的数只能是下标i之后的            

            tmp.pop_back(); // 递归返回到这可定是之前相等或者是大于了,所以要去掉tmp的最后一个,再继续找

        }

        if (num > target)

        {

            return;

        }

    }

}

vector<vector<int> > combinationSum(vector<int> &candidates, int target)

{

    std::sort(candidates.begin(), candidates.end()); // 因为给的数不是有序的,所以一定要先排好序,才能为之后避免重复解

    vector<vector<int> > ans;

    vector<int> tmp;

    back38(ans, candidates, tmp, target, 0);

    return ans;

}

};

这个在oj上要440ms。发现时间主要是用在计算vec的和上了。在网上看别人的做法,发现还有个巧妙的技巧。就是传入的target每次变小,知道为零是即为满足条件。剪枝判断是用target和当前想要加入的元素进行比较,如果target比较大那就把target减去要加入的元素之后的值当做下一次的target。同理如果要加入的值都比当前的target更大了就可以剪枝返回了。

这个代码只要80ms,提高了5倍

void back38(vector<vector<int> > &ans, vector<int> &candidates, vector<int> &tmp, int target, int cur)

{

    for (int i = cur; i < candidates.size(); ++i)

    {

        if (0 == target) // 为零即之前的组合满足最初target

        {

            ans.push_back(tmp);

            return;

        }

        if (candidates[i] <= target) // 如果将要加入的不比目标大,加入

        {

            tmp.push_back(candidates[i]);

            back38(ans, candidates, tmp, target - candidates[i], i); 

            tmp.pop_back();

        }

        if (candidates[i] > target) // 剪枝

        {

            return;

        }

    }

}

你可能感兴趣的:(LeetCode)