Question

40. Combination Sum II

中文文档

Description

Given a collection of candidate numbers (candidates) and a target number (target), find all unique combinations in candidates where the candidate numbers sum to target.

Each number in candidates may only be used once in the combination.

Note: The solution set must not contain duplicate combinations.

 

Example 1:

Input: candidates = [10,1,2,7,6,1,5], target = 8
Output: 
[
[1,1,6],
[1,2,5],
[1,7],
[2,6]
]

Example 2:

Input: candidates = [2,5,2,1,2], target = 5
Output: 
[
[1,2,2],
[5]
]

 

Constraints:

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

Solutions

Solution 1: Sorting + Pruning + Backtracking (Two Implementations)

We can first sort the array to facilitate pruning and skipping duplicate numbers.

Next, we design a function $dfs(i, s)$, which means starting the search from index $i$ with a remaining target value of $s$. Here, $i$ and $s$ are both non-negative integers, the current search path is $t$, and the answer is $ans$.

In the function $dfs(i, s)$, we first check whether $s$ is $0$. If it is, we add the current search path $t$ to the answer $ans$, and then return. If $i \geq n$ or $s \lt candidates[i]$, the path is invalid, so we return directly. Otherwise, we start the search from index $i$, and the search index range is $j \in [i, n)$, where $n$ is the length of the array $candidates$. During the search, if $j \gt i$ and $candidates[j] = candidates[j - 1]$, it means that the current number is the same as the previous number, we can skip the current number because the previous number has been searched. Otherwise, we add the current number to the search path $t$, recursively call the function $dfs(j + 1, s - candidates[j])$, and after the recursion ends, we remove the current number from the search path $t$.

We can also change the implementation logic of the function $dfs(i, s)$ to another form. If we choose the current number, we add the current number to the search path $t$, then recursively call the function $dfs(i + 1, s - candidates[i])$, and after the recursion ends, we remove the current number from the search path $t$. If we do not choose the current number, we can skip all numbers that are the same as the current number, then recursively call the function $dfs(j, s)$, where $j$ is the index of the first number that is different from the current number.

In the main function, we just need to call the function $dfs(0, target)$ to get the answer.

The time complexity is $O(2^n \times n)$, and the space complexity is $O(n)$. Here, $n$ is the length of the array $candidates$. Due to pruning, the actual time complexity is much less than $O(2^n \times n)$.

Similar problems:

• 39. Combination Sum
• 77. Combinations
• 216. Combination Sum III

class Solution:
  def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
    def dfs(i: int, s: int):
      if s == 0:
        ans.append(t[:])
        return
      if i >= len(candidates) or s < candidates[i]:
        return
      for j in range(i, len(candidates)):
        if j > i and candidates[j] == candidates[j - 1]:
          continue
        t.append(candidates[j])
        dfs(j + 1, s - candidates[j])
        t.pop()

    candidates.sort()
    ans = []
    t = []
    dfs(0, target)
    return ans

class Solution {
  private List<List<Integer>> ans = new ArrayList<>();
  private List<Integer> t = new ArrayList<>();
  private int[] candidates;

  public List<List<Integer>> combinationSum2(int[] candidates, int target) {
    Arrays.sort(candidates);
    this.candidates = candidates;
    dfs(0, target);
    return ans;
  }

  private void dfs(int i, int s) {
    if (s == 0) {
      ans.add(new ArrayList<>(t));
      return;
    }
    if (i >= candidates.length || s < candidates[i]) {
      return;
    }
    for (int j = i; j < candidates.length; ++j) {
      if (j > i && candidates[j] == candidates[j - 1]) {
        continue;
      }
      t.add(candidates[j]);
      dfs(j + 1, s - candidates[j]);
      t.remove(t.size() - 1);
    }
  }
}

class Solution {
public:
  vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
    sort(candidates.begin(), candidates.end());
    vector<vector<int>> ans;
    vector<int> t;
    function<void(int, int)> dfs = [&](int i, int s) {
      if (s == 0) {
        ans.emplace_back(t);
        return;
      }
      if (i >= candidates.size() || s < candidates[i]) {
        return;
      }
      for (int j = i; j < candidates.size(); ++j) {
        if (j > i && candidates[j] == candidates[j - 1]) {
          continue;
        }
        t.emplace_back(candidates[j]);
        dfs(j + 1, s - candidates[j]);
        t.pop_back();
      }
    };
    dfs(0, target);
    return ans;
  }
};

func combinationSum2(candidates []int, target int) (ans [][]int) {
  sort.Ints(candidates)
  t := []int{}
  var dfs func(i, s int)
  dfs = func(i, s int) {
    if s == 0 {
      ans = append(ans, slices.Clone(t))
      return
    }
    if i >= len(candidates) || s < candidates[i] {
      return
    }
    for j := i; j < len(candidates); j++ {
      if j > i && candidates[j] == candidates[j-1] {
        continue
      }
      t = append(t, candidates[j])
      dfs(j+1, s-candidates[j])
      t = t[:len(t)-1]
    }
  }
  dfs(0, target)
  return
}

function combinationSum2(candidates: number[], target: number): number[][] {
  candidates.sort((a, b) => a - b);
  const ans: number[][] = [];
  const t: number[] = [];
  const dfs = (i: number, s: number) => {
    if (s === 0) {
      ans.push(t.slice());
      return;
    }
    if (i >= candidates.length || s < candidates[i]) {
      return;
    }
    for (let j = i; j < candidates.length; j++) {
      if (j > i && candidates[j] === candidates[j - 1]) {
        continue;
      }
      t.push(candidates[j]);
      dfs(j + 1, s - candidates[j]);
      t.pop();
    }
  };
  dfs(0, target);
  return ans;
}

impl Solution {
  fn dfs(i: usize, s: i32, candidates: &Vec<i32>, t: &mut Vec<i32>, ans: &mut Vec<Vec<i32>>) {
    if s == 0 {
      ans.push(t.clone());
      return;
    }
    if i >= candidates.len() || s < candidates[i] {
      return;
    }
    for j in i..candidates.len() {
      if j > i && candidates[j] == candidates[j - 1] {
        continue;
      }
      t.push(candidates[j]);
      Self::dfs(j + 1, s - candidates[j], candidates, t, ans);
      t.pop();
    }
  }

  pub fn combination_sum2(mut candidates: Vec<i32>, target: i32) -> Vec<Vec<i32>> {
    candidates.sort();
    let mut ans = Vec::new();
    Self::dfs(0, target, &candidates, &mut vec![], &mut ans);
    ans
  }
}

/**
 * @param {number[]} candidates
 * @param {number} target
 * @return {number[][]}
 */
var combinationSum2 = function (candidates, target) {
  candidates.sort((a, b) => a - b);
  const ans = [];
  const t = [];
  const dfs = (i, s) => {
    if (s === 0) {
      ans.push(t.slice());
      return;
    }
    if (i >= candidates.length || s < candidates[i]) {
      return;
    }
    for (let j = i; j < candidates.length; ++j) {
      if (j > i && candidates[j] === candidates[j - 1]) {
        continue;
      }
      t.push(candidates[j]);
      dfs(j + 1, s - candidates[j]);
      t.pop();
    }
  };
  dfs(0, target);
  return ans;
};

public class Solution {
  private List<IList<int>> ans = new List<IList<int>>();
  private List<int> t = new List<int>();
  private int[] candidates;

  public IList<IList<int>> CombinationSum2(int[] candidates, int target) {
    Array.Sort(candidates);
    this.candidates = candidates;
    dfs(0, target);
    return ans;
  }

  private void dfs(int i, int s) {
    if (s == 0) {
      ans.Add(new List<int>(t));
      return;
    }
    if (i >= candidates.Length || s < candidates[i]) {
      return;
    }
    for (int j = i; j < candidates.Length; ++j) {
      if (j > i && candidates[j] == candidates[j - 1]) {
        continue;
      }
      t.Add(candidates[j]);
      dfs(j + 1, s - candidates[j]);
      t.RemoveAt(t.Count - 1);
    }
  }
}

Solution 2

class Solution:
  def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
    def dfs(i: int, s: int):
      if s == 0:
        ans.append(t[:])
        return
      if i >= len(candidates) or s < candidates[i]:
        return
      x = candidates[i]
      t.append(x)
      dfs(i + 1, s - x)
      t.pop()
      while i < len(candidates) and candidates[i] == x:
        i += 1
      dfs(i, s)

    candidates.sort()
    ans = []
    t = []
    dfs(0, target)
    return ans

class Solution {
  private List<List<Integer>> ans = new ArrayList<>();
  private List<Integer> t = new ArrayList<>();
  private int[] candidates;

  public List<List<Integer>> combinationSum2(int[] candidates, int target) {
    Arrays.sort(candidates);
    this.candidates = candidates;
    dfs(0, target);
    return ans;
  }

  private void dfs(int i, int s) {
    if (s == 0) {
      ans.add(new ArrayList<>(t));
      return;
    }
    if (i >= candidates.length || s < candidates[i]) {
      return;
    }
    int x = candidates[i];
    t.add(x);
    dfs(i + 1, s - x);
    t.remove(t.size() - 1);
    while (i < candidates.length && candidates[i] == x) {
      ++i;
    }
    dfs(i, s);
  }
}

class Solution {
public:
  vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
    sort(candidates.begin(), candidates.end());
    vector<vector<int>> ans;
    vector<int> t;
    function<void(int, int)> dfs = [&](int i, int s) {
      if (s == 0) {
        ans.emplace_back(t);
        return;
      }
      if (i >= candidates.size() || s < candidates[i]) {
        return;
      }
      int x = candidates[i];
      t.emplace_back(x);
      dfs(i + 1, s - x);
      t.pop_back();
      while (i < candidates.size() && candidates[i] == x) {
        ++i;
      }
      dfs(i, s);
    };
    dfs(0, target);
    return ans;
  }
};

func combinationSum2(candidates []int, target int) (ans [][]int) {
  sort.Ints(candidates)
  t := []int{}
  var dfs func(i, s int)
  dfs = func(i, s int) {
    if s == 0 {
      ans = append(ans, slices.Clone(t))
      return
    }
    if i >= len(candidates) || s < candidates[i] {
      return
    }
    for j := i; j < len(candidates); j++ {
      if j > i && candidates[j] == candidates[j-1] {
        continue
      }
      t = append(t, candidates[j])
      dfs(j+1, s-candidates[j])
      t = t[:len(t)-1]
    }
  }
  dfs(0, target)
  return
}

function combinationSum2(candidates: number[], target: number): number[][] {
  candidates.sort((a, b) => a - b);
  const ans: number[][] = [];
  const t: number[] = [];
  const dfs = (i: number, s: number) => {
    if (s === 0) {
      ans.push(t.slice());
      return;
    }
    if (i >= candidates.length || s < candidates[i]) {
      return;
    }
    const x = candidates[i];
    t.push(x);
    dfs(i + 1, s - x);
    t.pop();
    while (i < candidates.length && candidates[i] === x) {
      ++i;
    }
    dfs(i, s);
  };
  dfs(0, target);
  return ans;
}

impl Solution {
  fn dfs(mut i: usize, s: i32, candidates: &Vec<i32>, t: &mut Vec<i32>, ans: &mut Vec<Vec<i32>>) {
    if s == 0 {
      ans.push(t.clone());
      return;
    }
    if i >= candidates.len() || s < candidates[i] {
      return;
    }
    let x = candidates[i];
    t.push(x);
    Self::dfs(i + 1, s - x, candidates, t, ans);
    t.pop();
    while i < candidates.len() && candidates[i] == x {
      i += 1;
    }
    Self::dfs(i, s, candidates, t, ans);
  }

  pub fn combination_sum2(mut candidates: Vec<i32>, target: i32) -> Vec<Vec<i32>> {
    candidates.sort();
    let mut ans = Vec::new();
    Self::dfs(0, target, &candidates, &mut vec![], &mut ans);
    ans
  }
}

/**
 * @param {number[]} candidates
 * @param {number} target
 * @return {number[][]}
 */
var combinationSum2 = function (candidates, target) {
  candidates.sort((a, b) => a - b);
  const ans = [];
  const t = [];
  const dfs = (i, s) => {
    if (s === 0) {
      ans.push(t.slice());
      return;
    }
    if (i >= candidates.length || s < candidates[i]) {
      return;
    }
    const x = candidates[i];
    t.push(x);
    dfs(i + 1, s - x);
    t.pop();
    while (i < candidates.length && candidates[i] === x) {
      ++i;
    }
    dfs(i, s);
  };
  dfs(0, target);
  return ans;
};

public class Solution {
  private List<IList<int>> ans = new List<IList<int>>();
  private List<int> t = new List<int>();
  private int[] candidates;

  public IList<IList<int>> CombinationSum2(int[] candidates, int target) {
    Array.Sort(candidates);
    this.candidates = candidates;
    dfs(0, target);
    return ans;
  }

  private void dfs(int i, int s) {
    if (s == 0) {
      ans.Add(new List<int>(t));
      return;
    }
    if (i >= candidates.Length || s < candidates[i]) {
      return;
    }
    int x = candidates[i];
    t.Add(x);
    dfs(i + 1, s - x);
    t.RemoveAt(t.Count - 1);
    while (i < candidates.Length && candidates[i] == x) {
      ++i;
    }
    dfs(i, s);
  }
}