Leetcode:Combinations

链接

https://leetcode.com/problems/combinations/#/description

题目描述

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],
]

解题思路：

主要思想就是dfs.我们用list保存一种情况下的所有元素，用result保存最终的返回结果。当list中元素到达k个则找到一种情况放到result中，在index位置上替换为下一个数字进行dfs,代码如下：

注意：因为我们dfs过程中会将元素放入list中，所以在dfs结束之前将放入的元素删除。

代码如下：

public class Solution {
    private List<List<Integer>> result = new ArrayList<>();
    private int k;
    private int n;

    public List<List<Integer>> combine(int n, int k) {
        this.k = k;
        this.n = n;
        result.clear();
        List<Integer> list = new ArrayList<>();

        for(int i=1;i<=n-k+1;i++){
            combineHelper(i,list);
        }
        return result;
    }

    private void combineHelper(int index,List<Integer> list){
        list.add(index);
        if(list.size()==k){
            result.add(new ArrayList<>(list));
        }else{
            int size = list.size();
            for(int i=index+1;i<=n&&size+n-index>=k;i++){
                combineHelper(i,list);
            }
        }
        list.remove(list.size()-1);
        return;
    }
}