[leetcode每日一题2020/9/8]77. 组合

组合

  • 题目
  • 思路
  • 代码
  • 算法复杂度

题目来源于leetcode,解法和思路仅代表个人观点。传送门。
难度:中等
用时:大概40min

题目

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

示例:

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

思路

对于n=4,k=2这个例子
我们可以通过合理的枚举,输出所有可能的组合形式

[leetcode每日一题2020/9/8]77. 组合_第1张图片

枚举1~4中 k=2的1的可能所有组合
就等于
枚举2~4中 k=1的所有可能组合 再加上 1

抽象化,
枚举a~b中k位数的组合(结果)
就等于
枚举a~b中1的k位数组合 + 2的k位数组合 + … + b-k+1的k位数组合(因为只往后取,所以是不重复的)
其中 ,枚举x的k位数组合,就等于
枚举 (x+1~b的k-1位数组合) + x

代码

class Solution {
    //k<=n
    public List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> ans = new ArrayList<>();
        for(int i=1;i<=n-k+1;i++){
            ans.addAll(dp(i,n,k));
        }
        return ans;
    }
    public List<List<Integer>> dp(int a,int b,int k){
        if(k==1){
            List<List<Integer>> fin = new ArrayList<>();
            List<Integer> list = new ArrayList<>();
            list.add(a);
            fin.add(list);
            return fin;
        }
        List<List<Integer>> result = new ArrayList<>();
        for(int i=a;i<=b-k+1;i++) {
        	result.addAll(dp(i+1,b,k-1));
        }
        //把当前数加上
        for(int i=0;i<result.size();i++) {
        	result.get(i).add(a);
        }
        return result;
    } 
}

算法复杂度

时间复杂度: O(k*C(n,k))
空间复杂度: O(n)
[leetcode每日一题2020/9/8]77. 组合_第2张图片

你可能感兴趣的:(算法)