题目来源于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这个例子
我们可以通过合理的枚举,输出所有可能的组合形式
枚举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)