天天看点

leetcode-----回溯算法---组合

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

示例:

输入: n = 4, k = 2

输出:

[

[2,4],

[3,4],

[2,3],

[1,2],

[1,3],

[1,4],

]

这一题是典型的回溯算法,如果要求K个数的结果,我们可以通过求K-1个数的结果,然后在每个结果的后面加上一个比K-1个数的末尾还大的元素

public List<List<Integer>> combine(int n, int k) {
    	int i,j;
        List<List<Integer>> ans=new ArrayList<List<Integer>>();
        if(k==1) {
        	for(i=1;i<=n;i++) {
        		List<Integer> list1=new ArrayList<Integer>();
        		list1.add(i);
        		ans.add(list1);
        	}
        }
        else {
        	List<List<Integer>> pre=combine(n,k-1);
        	for(List<Integer> list1:pre) {
        		int end=list1.get(k-2);
        		for(j=end+1;j<=n;j++) {
        			List<Integer> list2=new ArrayList<Integer>();
        			list2.addAll(list1);
        			list2.add(j);
        			ans.add(list2);
        		}
        	}
        }
        return ans;
    }