[2018腾讯校园招聘模拟笔试编程题]小Q的歌单(python实现)

链接：小Q的歌单
来源：牛客网

小Q有X首长度为A的不同的歌和Y首长度为B的不同的歌，现在小Q想用这些歌组成一个总长度正好为K的歌单，每首歌最多只能在歌单中出现一次，在不考虑歌单内歌曲的先后顺序的情况下，请问有多少种组成歌单的方法。

输入描述:
每个输入包含一个测试用例。
每个测试用例的第一行包含一个整数，表示歌单的总长度K(1<=K<=1000)。
接下来的一行包含四个正整数，分别表示歌的第一种长度A(A<=10)和数量X(X<=100)以及歌的第二种长度B(B<=10)和数量Y(Y<=100)。保证A不等于B。

输出描述:

输出一个整数,表示组成歌单的方法取模。因为答案可能会很大,输出对1000000007取模的结果。
示例1
输入

5
2 3 3 3

输出

9

我的答案:

#计算排列组合数
def comb(n,m):
    up = 1
    for i in range(n):
        up = up * (m-i)
    down = 1
    for i in range(n):
        down = down * (i+1)
    return up//down

#接收输入
import sys
k = int(sys.stdin.readline())
line = sys.stdin.readline()
a, x, b, y = list(map(int,line.split()))

#遍历
res = 0
for i in range(x+1):
    for j in range(y+1):
        if (i * a + j * b) == k:
            res = res + comb(i,x)*comb(j,y)
#取模打印
print(res%1000000007)

从去年开始刷题,这道题也不是第一次见了.题目本身没什么难点,但模拟笔试时没有做出来.
得到的经验就是笔试题目不要想得太复杂,拿到题目后大概可以看出时间复杂度为O(n^2)的初步算法,就应该尝试编程.如果说题目分为算法构思和编程实现,对于有限时间的笔试而言,早早进入编程实现才是最实际的,毕竟最终打分也是按正确率,应当先尝试穷举解出,如果超时再重新想办法.