LeetCode40. 组合总和 II

题目

给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用一次。

说明:

  • 所有数字(包括目标数)都是正整数。
  • 解集不能包含重复的组合。 

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
  [1, 7],
  [1, 2, 5],
  [2, 6],
  [1, 1, 6]
]

示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
  [1,2,2],
  [5]
]

分析

用回溯把所有集合都遍历一遍,因为有重复元素,所以得到的集合会有重复,比如1 7 是一个集合,之后还会可能有一个7 1。

去重的话看了别人的解题报告,是这样做的:先把数组排了序,方便后续操作,对于重复元素,比如说 1 2 2 2 5 target = 5

当重复元素2参加了回溯之后,剩下的2就不进入回溯中,直接跳过。详细看代码吧。

代码

class Solution {
    public List> combinationSum2(int[] candidates, int target) {
        Arrays.sort(candidates);
        List> list = new ArrayList<>();
        List listTemp = new ArrayList<>();
        combination(list,listTemp,candidates,target,0);
        return list;
    }
    
    
    public static void combination(List> list, List listTemp, int[] candidates, int target, int index){

        if (target == 0){
            System.out.println(listTemp);
            list.add(new ArrayList<>(listTemp));
            return;
        }

        for (int i = index; i < candidates.length; i++) {
            if (candidates[i] <= target){
                listTemp.add(candidates[i]);
                target -= candidates[i];
                combination(list,listTemp,candidates,target,i+1);
                listTemp.remove(listTemp.size()-1);
                target += candidates[i];
                //跳过重复元素
                while (i < candidates.length-1 && candidates[i] == candidates[i+1]){
                    i++;
                }
            }
        }
    }
}

 

你可能感兴趣的:(算法)