【代码随想录21】77.组合

最新推荐文章于 2024-11-02 15:25:13 发布

码农阿祖

最新推荐文章于 2024-11-02 15:25:13 发布

阅读量362

点赞数 7

分类专栏：代码随想录文章标签：算法 java 数据结构

本文链接：https://blog.csdn.net/m0_50521311/article/details/136001454

版权

代码随想录专栏收录该内容

47 篇文章 0 订阅

订阅专栏

本文介绍了一个编程问题，要求在给定整数n和k的条件下，找出所有可能的由1到n组成的k个数的组合。使用深度优先搜索(DFS)和回溯策略来解决此问题，给出了Solution类中的关键函数和示例输出。

摘要由CSDN通过智能技术生成

在这里插入图片描述

77.组合

题目描述

给定两个整数 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

参考代码

class Solution {
    List<Integer> temp = new ArrayList<Integer>();
    List<List<Integer>> ans = new ArrayList<List<Integer>>();

    public List<List<Integer>> combine(int n, int k) {
        dfs(1, n, k);
        return ans;
    }

    public void dfs(int cur, int n, int k) {
        // 剪枝：temp 长度加上区间 [cur, n] 的长度小于 k，不可能构造出长度为 k 的 temp
        if (temp.size() + (n - cur + 1) < k) {
            return;
        }
        // 记录合法的答案
        if (temp.size() == k) {
            ans.add(new ArrayList<Integer>(temp));
            return;
        }
        // 考虑选择当前位置
        temp.add(cur);
        dfs(cur + 1, n, k);
        temp.remove(temp.size() - 1);
        // 考虑不选择当前位置
        dfs(cur + 1, n, k);
    }
}