leetcode-40-组合总和 II

1,131 阅读1分钟

我正在参加「掘金·启航计划」
题目地址

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

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

注意:解集不能包含重复的组合。 

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8, 输出: [ [1,1,6], [1,2,5], [1,7], [2,6] ] 示例 2:

输入: candidates = [2,5,2,1,2], target = 5, 输出: [ [1,2,2], [5] ]

提示:

1 <= candidates.length <= 100 1 <= candidates[i] <= 50 1 <= target <= 30

解题思路

本题解题思路并不难,就是求给定集合的全排列,并从中筛选出和值为 target 的组合。所谓全排列,就是所有可能的组合,以 [1,2,3] 为例,它的全排列有 1,2,3,12,13,23,123
难点在于本题对时间复杂度要求高一些,所以我们要尽可能高效的找出符合题意的组合,那么此时,我们就要利用题目中给出的信息减少不必要的组合查找。
本题要求解集中不能包含重复的组合,同时,给定数组中每个数都是大于等于 1 的,有了这些信息,我们就可以想到此类题常用的一个技巧,先对输入数组排序,这里我们按升序排序。
排序后,我们首先遍历一次输入数组,把所有大于 target 的数字截取掉,因为这些数字不可能和其他数字组成和值为 target 的组合,输入数组越小,我们后边进行全排列的时候处理的次数就越少,同时这一步可以把所有等于 target 的数字找出来。
接下来,我们需要封装一个函数 getMore,该函数接收输入数组,开始下标,本次集合,已累计和值,和目标值,它的功能就是基于现有集合继续向后匹配每一个数字,看是否可以凑合符合题意的组合。
这里有几个可以提高效率的点:

  1. 在解题最开始,我们已经把输入数组升序排序了,而且本题要求不能有重复的组合,那么每次遍历调用 getCombination 的过程中,如果当前数字和前一个数字相同,则当前数字就可以跳过了,因为哪怕它可以找到和值为 target 的组合,那也必然和前一个数字找到的相同
  2. 如果当前找到的组合和值已经大于 target 了,那么就没有必要继续向后查找了,因为后面的数字更大,找到了组合和值必然也更大,必然不会有符合要求的组合

以上两点就体现出了最开始把输入数组升序排序的重要性了。

代码实现

function getCombination(candidates, start, combination, sum, target) {
  const result = []

  for (let i = start; i < candidates.length; i++) {
    if (i > start && candidates[i - 1] === candidates[i]) {
      continue
    }

    const _sum = sum + candidates[i]
    if (_sum < target) {
      result.push(...getCombination(candidates, i + 1, [...combination, candidates[i]], _sum, target))
    } else if (_sum === target) {
      result.push([...combination, candidates[i]])
    } else {
      break
    }
  }

  return result
}

var combinationSum2 = function (candidates, target) {
  const result = []

  candidates.sort((a, b) => a - b)

  for (let i = 0; i < candidates.length; i++) {
    if (candidates[i] === target && !result.includes(target)) {
      result.push([target])
      candidates.splice(i + 1)
    }
    if (candidates[i] > target) {
      candidates.splice(i)
    }
  }

  for (let i = 0; i < candidates.length; i++) {
    if (i > 0 && candidates[i - 1] === candidates[i]) {
      continue
    }
    result.push(...getCombination(candidates, i + 1, [candidates[i]], candidates[i], target))
  }

  return result
}

拿下!😀