77. Combinations (M)

最新推荐文章于 2020-12-19 13:08:16 发布

原创最新推荐文章于 2020-12-19 13:08:16 发布 · 120 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#回溯法

LeetCode 专栏收录该内容

202 篇文章

订阅专栏

本文探讨了从n个数中选择k个数的所有可能组合。通过使用回溯法，有效地解决了组合问题，并提供了详细的代码实现。文章重点介绍了算法的思路和剪枝技巧。

Combinations (M)

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个组成集合，输出所有这样的集合。

思路

回溯法，注意剪枝。

代码实现

class Solution {
    public List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> ans = new ArrayList<>();
        combine(n, k, 1, new ArrayList<>(), ans);
        return ans;
    }

    private void combine(int n, int k, int cur, List<Integer> list, List<List<Integer>> ans) {
        if (list.size() == k) {
            ans.add(new ArrayList<>(list));
            return;
        }

        // 只有当剩余元素数加上已加入列表元素数大于等于k时，才需要继续递归
        for (int i = cur; n - i + 1 >= k - list.size(); i++) {
            list.add(i);
            combine(n, k, i + 1, list, ans);
            list.remove(list.size() - 1);
        }
    }
}