前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >77. Combinations「建议收藏」

77. Combinations「建议收藏」

作者头像
全栈程序员站长
发布2022-11-08 14:44:29
2460
发布2022-11-08 14:44:29
举报
文章被收录于专栏:全栈程序员必看

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:

代码语言:javascript
复制
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

题意:给定n和k,求1-n中任意k个数的组合。

这个题我们给出2种解法

第一种:递归

对于任意的n和k,

1.我们先求出从1到n-1中任意k个数的组合

2.然后求,1到n-1中任意k-1个数的组合,每个组合加n,

最后将组合合并到一个集合即可

具体代码:

代码语言:javascript
复制
class Solution {
    public List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> list = new ArrayList<>();
        if(n<k)
            return list;
        if(n==k){
            List<Integer> list1 = new ArrayList<>();
            for(int i=1;i<=n;i++)
                list1.add(i);
            list.add(list1);
            return list;
        }
        if(k == 1){
            for(int i=0;i<n;i++){
                List<Integer> list1 = new ArrayList<>();
                list1.add(i+1);
                list.add(list1);
            }
            return list;
        }
        List<List<Integer>> list2 = combine(n-1,k-1);
        for(List<Integer> l : list2){
            l.add(n);
        }
        list = combine(n-1,k);
        list.addAll(list2);
        return list;
    }
}

第二种解法,我们使用回溯的思想

代码语言:javascript
复制
class Solution {
    public List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> list = new ArrayList<>();
          if(n<k)
            return list;
        combine(list,new ArrayList<Integer>(),k,n,1);
        return list;
    }
    public void combine(List<List<Integer>> list,List<Integer> tempList,int k,int n,int start){
        if(tempList.size() == k){
            list.add(new ArrayList<Integer>(tempList));
        }else{
            for(int i=start;i<=n;i++){
                tempList.add(i);
                combine(list,tempList,k,n,i+1);
                tempList.remove(tempList.size()-1);
            }
        }
    }
}

上述2种解法中,递归解法在LeetCode上超越了90%左右,而使用回溯的解法只超越了40%多,显然递归要好一些。对此,我觉得可能是使用递归的过程中计算的都是我们需要求的组合,但是在回溯中会多次出现碰到“死胡同”这样的情况,导致了2种解法在时间空间上存在差异。举个例子:

我们要求n=9,k=9的组合,显然只有一种组合满足题意,但是当我们使用回溯算法时,tempList分别加入1,2,3,4,5,6,7,8,9之后就不需要再计算了,但事实上,我们仍旧还计算了加入tempList中的第一个元素为2,3,4,5,6,7,8,9的情况,而这些情况显然已经不符合题意了,我觉得这个过程做的无用功是导致这种解法不优的原因。

既然如此,我们在原回溯解法上稍作优化,在回溯方法的第一行加上一个判断条件

代码语言:javascript
复制
if(tempList.size()+n-start+1 < k)
            return;

假设之后所有元素都加入tempList中都比k小的时候,我们不需要再计算了。

这样一来,运行结果在LeetCode上超越了85%左右,与递归解法相差小了很多;

由此可见,有时候一行代码足以改变一种算法的优劣程度

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/185263.html原文链接:https://javaforall.cn

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2022年10月6日 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档