给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]
思路:回溯算法。
class Solution {
public List<List<Integer>> combine(int n, int k) {
List<List<Integer>> res=new LinkedList<List<Integer>>();
List<Integer> tmp=new LinkedList<Integer>();
int nums[]=new int[n];
for(int i=0;i<n;i++){
nums[i]=i+1;
}
boolean used[]=new boolean[n];
dfs(res,tmp,nums,used,k,0);
return res;
}
public void dfs(List<List<Integer>> res,List<Integer> tmp,int[] nums,boolean[] used,int k,int index){
if(tmp.size()==k){
res.add(new LinkedList<>(tmp));
return ;
}
for(int i=index;i<nums.length;i++){
if(used[i]!=true){
tmp.add(nums[i]);
used[i]=true;
dfs(res,tmp,nums,used,k,i+1);
used[i]=false;
tmp.remove(tmp.size()-1);
}
}
}
}