迅雷18年笔试中的一题

最新推荐文章于 2021-03-04 23:42:16 发布

hsc_1

最新推荐文章于 2021-03-04 23:42:16 发布

阅读量271

点赞数

分类专栏：算法 python

本文链接：https://blog.csdn.net/hsc_1/article/details/82908052

版权

python 同时被 2 个专栏收录

143 篇文章 1 订阅

订阅专栏

算法

57 篇文章 0 订阅

订阅专栏

给定整数n，取若干个1到n的整数可求和等于整数m，编程求出所有组合的个数。比如当n=6，m=8时，有四种组合：[2,6], [3,5], [1,2,5], [1,3,4]。限定n和m小于120

输入描述:

整数n和m

输出描述:

求和等于m的所有组合的个数。

输入例子1:

6 8

输出例子1:

没在牛客网上答过题目，第一次在牛客网上答题是头条的笔试。

很是崩溃，我他妈连输入都不知道怎么获取，更何况输出怎么给了！！！

value = input()
n, m = [int(ele) for ele in value.split(' ')]
def func(n, m):
    nums = [[0 for _ in range(m + 1)] for _ in range(n+1)]
    for i in range(n+1):
        nums[i][0] = 1
    for i in range(1, n+1):
        for j in range(1, m+1):
            if i > j :
                nums[i][j] = nums[i-1][j]
            else:
                nums[i][j] = nums[i-1][j] + nums[i-1][j-i]
    print(nums[-1][-1])
func(n, m)