77. 组合 - 力扣(LeetCode)

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

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

输入示例

n = 4, k = 2

输出示例

[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

解题思路
我们使用回溯、深度优先遍历的思想,我们使用一个栈 path 来记录走过的路径,使用 begin 来记录当前来到的数字位置。递归过程如下:

  • 如果 path 的长度等于 k,表示已经组合成功
  • 从 begin 开始遍历所有可能的搜索起点(不是从 1 开始,因为当前位置之前的数字已经走过),以下是遍历内操作:
  • 向路径变量里添加一个数
  • 下一轮搜索,设置起点要加 1,因为组合里不允许出现重复的元素
  • 深度优先遍历有回头过程,递归之后进行逆操作
    77. 组合 - 力扣(LeetCode)_第1张图片

解题代码

class Solution {
    public List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> ans = new ArrayList<>();
        if(k <= 0 || n < k) {
            return ans;
        }
        // 从1开始是题目的设定
        Deque<Integer> path = new ArrayDeque<>();
        backtrack(ans, path, 1, n, k);
        return ans;
    }

    public void backtrack(List<List<Integer>> ans, Deque<Integer> path, int begin, int n, int k) {
        // 递归终止条件是path的长度等于k
        if(path.size() == k) {
            ans.add(new ArrayList<>(path));
            return;
        }
        // 遍历可能的搜索起点
        for(int i = begin; i <= n; i++) {
            // 向路径变量里添加一个数
            path.addLast(i);
            // 下一轮搜索,设置起点要加1,因为组合数里不允许出现重复的元素
            backtrack(ans, path, i+1, n, k);
            // 深度优先遍历有回头过程,递归之后做逆向操作
            path.removeLast();
        }
    }
}

你可能感兴趣的:(leetcode,算法,职场和发展)