[LeetCode] 77. Combinations

Posted 2020-12-23 CNoodle

Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.

Example:

Input: n = 4, k = 2
Output:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

组合。题意是给一个数字N和一个数字K，请你返回 1 ... n 中所有可能的 k 个数的组合。依然是backtracking类的题目，注意这里N的下限是从1开始。同时这个题又规定了combination的数字个数只能为K个所以helper函数的退出条件是K == 0。

时间O(n^min(k, n - k))

空间O(n)

Java实现

 1 class Solution {
 2     public List<List<Integer>> combine(int n, int k) {
 3         List<List<Integer>> res = new ArrayList<>();
 4         helper(res, new ArrayList<>(), n, k, 1);
 5         return res;
 6     }
 7 
 8     private void helper(List<List<Integer>> res, List<Integer> list, int n, int k, int start) {
 9         // 退出条件，当res里的个数满足K了，就加入结果集
10         if (k == 0) {
11             res.add(new ArrayList<>(list));
12             return;
13         }
14         // 选取一个数字作为组合的第一个数字，直到N
15         for (int i = start; i <= n; i++) {
16             list.add(i);
17             // 递归的下一层里面K就减1，但是同时开始的index也需要 + 1
18             helper(res, list, n, k - 1, i + 1);
19             list.remove(list.size() - 1);
20         }
21     }
22 }

 

同时我附上这个帖子，根据起点画出的二叉树对于解决这个问题很有帮助。

 

LeetCode 题目总结