给定两个整数 n 和 k,返回 1 … n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
其实就是Cnk的组合问题,递归函数和参数从来都是重点,设计一个idx来作为指针,因为要取k个数,因此当k==0的时候作为边界条件,还要加上idx扫描越界和k小于零的情况,因此重点就是递归函数,由于是不重复取,所以用boolean数组记录一下两种决策。
class Solution {
List<List<Integer>> ans=new ArrayList<>();
boolean [] v=new boolean[100];
public void robot(int idx,int n,int k){
if(k==0){
//拼答案
List<Integer> res=new ArrayList<>();
for(int i=0;i<n;i++){
if(v[i])
res.add(i+1);
}
ans.add(res);
return;
}
if(idx>=n||k<0)
return;
v[idx]=true;
robot(idx+1,n,k-1);//选这个数字
v[idx]=false;
robot(idx+1,n,k);//不选这个数字
}
public List<List<Integer>> combine(int n, int k) {
robot(0,n,k);
return ans;
}
}