Code Ganker: Combination Sum II -- LeetCode

2014年3月8日星期六

Combination Sum II -- LeetCode

原题链接: http://oj.leetcode.com/problems/combination-sum-ii/
这道题跟Combination Sum非常相似,不了解的朋友可以先看看,唯一的区别就是这个题目中单个元素用过就不可以重复使用了。乍一看好像区别比较大,但是其实实现上只需要一点点改动就可以完成了,就是递归的时候传进去的index应该是当前元素的下一个。代码如下:
public ArrayList<ArrayList<Integer>> combinationSum2(int[] num, int target) {
    ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
    if(num == null || num.length==0)
        return res;
    Arrays.sort(num);
    helper(num,0,target,new ArrayList<Integer>(),res);
    return res;
}
private void helper(int[] num, int start, int target, ArrayList<Integer> item,
ArrayList<ArrayList<Integer>> res)
{
    if(target == 0)
    {
        res.add(new ArrayList<Integer>(item));
        return;
    }
    if(target<0 || start>=num.length)
        return;
    for(int i=start;i<num.length;i++)
    {
        if(i>start && num[i]==num[i-1]) continue;
        item.add(num[i]);
        helper(num,i+1,target-num[i],item,res);
        item.remove(item.size()-1);
    }
}
在这里我们还是需要在每一次for循环前做一次判断,因为虽然一个元素不可以重复使用,但是如果这个元素重复出现是允许的,但是为了避免出现重复的结果集,我们只对于第一次得到这个数进行递归,接下来就跳过这个元素了,因为接下来的情况会在上一层的递归函数被考虑到,这样就可以避免重复元素的出现。这个问题可能会觉得比较绕,大家仔细想想就明白了哈。

3 条评论:

  1. 如果在helper的for 循环里的最后一行加个判断,如果target-num[i]<=0 , 就break 这个循环,或者直接return, 感觉上时间上和空间上都能省一点点。上一道题也一样。楼主看一下是不是。

    回复删除
    回复
    1. 谢谢你的回复哈,是的,这样是可以省一点时间和空间,不过其实并不明显,因为即使不在这里return,递归到下一层target会小于0,所以也会直接return,也就省一层递归调用而已。一般来说递归有两种规则,一种就是判了再进,另一种是进了再判,一般来说认为这两种是差不多的,只是前者省一点递归调用。

      删除
    2. peng的意思是应该用了break之后就不用check那一层的兄弟了, 是一个剪枝的作用
      一看时间过去2年了....

      删除