小Q的歌单

TX2018秋招笔试题

#include<iostream>
using namespace std;
const int mod = 1000000007;
long long c[105][105];
//初始组合数。比如c[4][2]，就是combination(4,2)的意思，等于6
void init()
{
    c[0][0] = 1;
    for(int i =1;i<=100;++i)
    {
        c[i][0] = 1;
        for(int j = 1;j<=100;++j)
        {
            c[i][j] = (c[i-1][j-1] + c[i-1][j])%mod;
        }
    }
}
 
int main()
{
    int k,a,b,x,y;
    long long ans = 0;
    init();
    scanf("%d",&k);
    scanf("%d%d%d%d",&a,&x,&b,&y);
    //对长度为A的歌单，假如从A中选了i首，检查剩余的长度在B中能不能选
    for(int i =0;i<=x;++i)
    {
        if(i*a <=k && (k-i*a)%b==0 && (k-i*a)/b<=y)
        {//求组合数，x首中选i首，和y首中选(k-i*a)/b首，相乘
            ans = (ans+(c[x][i]*c[y][(k-i*a)/b])%mod)%mod;
        }
    }
    printf("%lld\n",ans);
    return 0;
}