Given two integers n and k, return all possible combinations of k numbers out of 1 ... n. For example,If n = 4 and k = 2, a solution is: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4],] Subscribe to see which companies asked this question.

题目

组给出两个整数n和k，返回从1......n中选出的k个数的组合。 样例 例如 n = 4 且 k = 2 返回的解为： [[2,4],[3,4],[2,3],[1,2],[1,3],[1,4]]

分析

典型的深度搜索回溯问题

代码

public class Solution {
    /**
     * @param n: Given the range of numbers
     * @param k: Given the numbers of combinations
     * @return: All the combinations of k numbers out of 1..n
     */
    public ArrayList<ArrayList<Integer>> combine(int n, int k) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        
        ArrayList<Integer> list = new ArrayList<>();
        
        if(n < k)
        	return res;
        
        dfs(res, list, n, k, 1);
        
        return res;
    }

	private void dfs(ArrayList<ArrayList<Integer>> res, ArrayList<Integer> list, int n, int k, int start) {
		
		if(list.size() == k) {
			res.add(new ArrayList<>(list));
			return;
		}
		
		for(int i=start;i<=n;i++) {
			if(!list.contains(i))
				list.add(i);
			dfs(res, list, n, k, i+1);
			list.remove(list.size()-1);
		}
	}
}