javascript - 有一个数组，从中挑ABC三个整数，让ABC三个整数加起来等于0，看有多少个这样的数组？-PHP中文网问答

javascript - 有一个数组，从中挑ABC三个整数，让ABC三个整数加起来等于0，看有多少个这样的数组？

高洛峰 2017-04-11 12:13:50

[JavaScript讨论组]

275

面试题，求解答

高洛峰

拥有18年软件开发和IT教学经验。曾任多家上市公司技术总监、架构师、项目经理、高级软件工程师等职务。网络人气名人讲师，...

全部回复(3)

怪我咯2017-04-11 12:15:50 3楼

都确定是3个整数了……直接三重循环枚举啊…………

赞 +0

添加回复

伊谢尔伦2017-04-11 12:15:50 2楼

如同 @xiaoboost 說的, 使用簡單的暴力法就可以做出來:

import itertools

def nsum(lst, n, target):
    count = 0
    for c in itertools.combinations(lst, n):
        if sum(c) == target:
            count += 1
    return count


if __name__ == '__main__':
    lst = [-1, 3, -2, 7, -4, -3, 6, 9, -2, -2, 1, 1, 0, 10, -1, 9, 8, -12]
    print(nsum(lst, 3, 0))

結果:

我回答過的問題: Python-QA

赞 +0

添加回复

天蓬老师2017-04-11 12:15:50 1楼

from itertools import combinations

def sum_is_sth(lst, sth=0, cnt=3):
    return sum([1 for sub_lst in combinations(lst, cnt) if sum(sub_lst) == sth])

赞 +0

添加回复

专题推荐

更多>

热门话题

热门教程

更多>