LeetCode笔记——77组合

题目:

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

示例:

输入: n = 4, k = 2
输出:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

思路:以下参考了网上大神的代码,原文链接:https://blog.csdn.net/happyaaaaaaaaaaa/article/details/51564160

是一个递归实现的问题,有一点深度优先搜索的意思。递归的返回条件是k==0,即找到了长度为K的一个组合,将其加入到结果集中,并且在temp中删除。

代码:

class Solution {
    public List> combine(int n, int k) {
     List> res = new ArrayList>();
        List temp = new ArrayList();
        dfs(res, temp, n, k, 1);
        return res;
    }
    private void dfs(List> res, List temp, int n, int k, int m) {
        if(k == 0) {  //递归结束条件
            res.add(new ArrayList(temp));
            return;
        }
        for(int i=m; i<=n; i++) {
            temp.add(i);    //将第一个元素添加进temp
            dfs(res, temp, n, k-1, i+1);   //递归调用,从i+1开始,避免重复
            temp.remove(temp.size()-1); //取出原先存储的组合,重新开始
        }

    }
}

以下这两个思路一致,只不过多加了一个判断条件。

提交最多的代码:

class Solution {
    public List> combine(int n, int k) {
        List> res = new ArrayList>();
        if(k>n) return res;
        List list = new ArrayList();
        getPass(n, k, list, res, 1);
        return res;
    }
    
    public void getPass(int n, int k, List list, List> res, int start){
        if(list.size() == k){
            res.add(new ArrayList(list));
            return;
        }
        int len = list.size();
        for(int i=start;i<=n;i++){
            if(n-i<(k-len-1)) return;
            list.add(i);
            getPass(n, k, list, res, i+1);
            list.remove(list.size()-1);
        }
    }
}

执行最快的代码:

class Solution {
    public List> combine(int n, int k) {
        List> res = new ArrayList>();
        if(k>n) return res;
        List list = new ArrayList();
        getPass(n, k, list, res, 1);
        return res;
    }
    
    public void getPass(int n, int k, List list, List> res, int start){
        if(list.size() == k){
            res.add(new ArrayList(list));
            return;
        }
        int len = list.size();
        for(int i=start;i<=n;i++){
            if(n-i<(k-len-1)) return;
            list.add(i);
            getPass(n, k, list, res, i+1);
            list.remove(list.size()-1);
        }
    }
}

你可能感兴趣的:(LeetCode笔记)