迅雷18年笔试中的一题

版权声明： https://blog.csdn.net/hsc_1/article/details/82908052

给定整数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:

4

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

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

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)