腾讯笔试题 小Q的歌单

/*组合数学解法
*（1）杨辉三角中的值对应于其所在位置的组合数，例如C[4][2]，表示从4个里面选2个，等于6
*（2）当满足i首长度为A的歌小于总长度K，且满足K减去i乘以A后对长度为B的歌取模的值为0，且满
*足K减去i乘以A后整除B小于等于Y后，即(i*A<=K) && (K-i*A)%B==0 && ((K-i*A)/B)<=Y
*进行组合，从X首中选择i首，从Y首中选择(K-i*A)/B首，然后相乘
*/
#include <bits/stdc++.h>
using namespace std;
const int maxn = 105;
const int mod = 1000000007;
long long c[maxn][maxn];
void init()
{
	c[0][0] = 1;
	for (int i=1; i<maxn; i++)
	{
		c[i][0] = 1;
		for (int j=1; j<maxn; 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);
	for (int i=0; i<=X; i++)
	{
		if( (i*A<=K) && (K-i*A)%B==0 && ((K-i*A)/B)<=Y )
		{
			ans = (ans + (c[X][i]*c[Y][(K-i*A)/B]) % mod) % mod;
		}
	}
	printf ("%lld", ans);
	return 0;
}

/*动态规划解0-1背包问题解决方案
*（1）子问题的定义：dp[i][j]表示从i个物品中选若干个物品，放入空间是j的背包中
*（2）最优解：根据第i个物品放与不放进行决策，可以推出两种状态。第一种是不能放，包
*就这么大，放不下，上一次的刚好；第二种是还能放，我用上一次放完空出来的空间，来
*放另外的物品
*/
#include <bits/stdc++.h>
using namespace std;
long long dp[201][1001];
long p[201];

int main()
{
	int K, A, X, B, Y;
	int mod = 1000000007;
	scanf ("%d", &K);
	scanf ("%d %d %d %d", &A, &X, &B, &Y);
	memset(dp, 0, sizeof(dp));
	dp[0][0] = 1;	
	for (int i=1; i<=X; i++)
		p[i] = A;
	for (int j=X+1; j<=X+Y; j++)
		p[j] = B;
	for (int i=1; i<=X+Y; i++)
		for (int j=0; j<=K; j++)
		{
			if (j >= p[i])
				dp[i][j] = (dp[i-1][j] % mod + dp[i-1][j-p[i]] % mod) % mod;
			else
				dp[i][j] = dp[i-1][j]%mod;
		}
	printf ("%lld", dp[X+Y][K]%mod);
	return 0;
}

/*参考博客：
*https://www.jianshu.com/p/48f2dd394608
*https://blog.csdn.net/yue_jijun/article/details/81105030
*/