多个数组,每个数组中选择一项,总共能有多少种组合,reduce函数实现

这篇博客探讨了如何使用reduce函数解决从多个数组中选择一项并计算所有可能组合的问题。作者提到先前使用treelib实现的方法虽然性能好,但内存消耗过大,可达30G。因此,他们转向使用reduce函数来优化内存使用。文章提供了测试结果,并鼓励读者查看代码以了解更多详情。
摘要由CSDN通过智能技术生成

问题来源:《有意思的面试题-数独问题》
我在《多个数组,每个数组中选择一项,总共能有多少种组合,treelib实现》中写了通过treelib实现的方式,但我发现这个方式内存消耗量极大,有的时候内存使用量达到30G,reduce对内存的使用也很大,但性能上胜出,所以这次使用reduce搞一把。
废话不多说了,看代码:

import numpy as np
import sys

def combination(lists):

    if sys.version_info < (3, 0):
        import reduce
    else:
        from functools import reduce

    def reduceFun(list1, list2):
        array = []
        for i in list1:
            for j in list2:
                array.append(np.hstack((i, j)))
        return array

    return reduce(reduceFun, lists)

#lists =  [[6, 8], [2, 5], [1, 9], [3, 9], [3, 5], [3, 9], [8, 9], [3, 6], [2, 9], [2, 3, 8], [2, 3, 6], [2, 5, 8], [1, 5, 8], [1, 6, 9], [5, 6, 9], [1, 3, 5], [3, 5, 6], [3, 8, 9]]
lists =  [[1, 2, 3], [4,<
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值