leetcode 77. Combinations 组合(中等)

129 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第11天,点击查看活动详情

一、题目大意

标签: 搜索

leetcode.cn/problems/co…

给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。

你可以按 任何顺序 返回答案。

示例 1:

输入:n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]

示例 2:

输入:n = 1, k = 1 输出:[[1]]

提示:

  • 1 <= n <= 20
  • 1 <= k <= n

二、解题思路

用回溯方法解决组合问题,类似排列,排列回溯的是交换的位置,而组合回溯的是否把当前的数字加入结果中。

三、解题方法

3.1 Java实现

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

    void backTracking(List<List<Integer>> ans, int[] comb, int count, int pos, int n, int k) {
        if (count == k) {
            ans.add(Arrays.stream(comb).boxed().collect(Collectors.toList()));
            return;
        }
        for (int i = pos; i <= n; i++) {
            // 修改当前节点状态
            comb[count++] = i;
            // 递归子节点
            backTracking(ans, comb, count, i + 1, n, k);
            // 回改当前节点状态
            count--;
        }
    }
}

四、总结小记

  • 2022/6/4 好吃的东西很多,要克制