Leetcode-77. Combinations

前言：为了后续的实习面试，开始疯狂刷题，非常欢迎志同道合的朋友一起交流。因为时间比较紧张，目前的规划是先过一遍，写出能想到的最优算法，第二遍再考虑最优或者较优的方法。如有错误欢迎指正。博主首发优快云，mcf171专栏。

博客链接：mcf171的博客

——————————————————————————————

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

这个题目想了想还是可以用递归来做，一个List保存当前的数，一个保存剩下的数。第一次过了，但是效果极差。 Your runtime beats 0.34% of java submissions.

public class Solution {
    public List<List<Integer>> combine(int n, int k) {

        List<List<Integer>> results = new ArrayList<List<Integer>>();
        List<Integer> currentResult = new ArrayList<Integer>();
        List<Integer> remain = new ArrayList<Integer>();
        for(int i = 1; i <= n;  i++) remain.add(i);

        generate(results,currentResult,remain,k);
        return results;

    }

    public void generate(List<List<Integer>> results,List<Integer> currentResult, List<Integer> remain, int k){

        if(k == 1){
            for(int item : remain){
                List<Integer> result = new ArrayList<Integer>(currentResult);
                result.add(item);
                results.add(result);
            }
        }else{
            int times = remain.size();
         
            for(int i = 0 ; i < times; i ++){
                int value = remain.get(0);
                currentResult.add(value);
                remain.remove(0);
                generate(results,currentResult,new ArrayList<Integer>(remain),k - 1);
                currentResult.remove(currentResult.size()-1);
            }
        }
    }
}

后来思考了一下，每次递归应该先检查是否符合继续递归的情况。 Your runtime beats 80.87% of java submissions.

public class Solution {
    public List<List<Integer>> combine(int n, int k) {

        List<List<Integer>> results = new ArrayList<List<Integer>>();
        List<Integer> currentResult = new ArrayList<Integer>();
        List<Integer> remain = new ArrayList<Integer>();
        for(int i = 1; i <= n;  i++) remain.add(i);

        generate(results,currentResult,remain,k);
        return results;

    }

    public void generate(List<List<Integer>> results,List<Integer> currentResult, List<Integer> remain, int k){

        if(k == 1){
            for(int item : remain){
                List<Integer> result = new ArrayList<Integer>(currentResult);
                result.add(item);
                results.add(result);
            }
        }else{
            int times = remain.size();
            if (k > times) return;
            for(int i = 0 ; i < times; i ++){
                int value = remain.get(0);
                currentResult.add(value);
                remain.remove(0);
                generate(results,currentResult,new ArrayList<Integer>(remain),k - 1);
                currentResult.remove(currentResult.size()-1);
            }
        }
    }
}