组合数学 (习题): HDU 2152

桃扇骨 2022-02-03 22:43 351阅读 0赞

HDU 2152

  • 题目
  • 输入输出
    • 输入
    • 输出
  • 思路
  • 代码 (C 语言)
  • 总结

题目

转眼到了收获的季节,由于有TT的专业指导,Lele获得了大丰收。特别是水果,Lele一共种了N种水果,有苹果,梨子,香蕉,西瓜……不但味道好吃,样子更是好看。

于是,很多人们慕名而来,找Lele买水果。

甚至连大名鼎鼎的HDU ACM总教头 lcy 也来了。lcy抛出一打百元大钞,“我要买由M个水果组成的水果拼盘,不过我有个小小的要求,对于每种水果,个数上我有限制,既不能少于某个特定值,也不能大于某个特定值。而且我不要两份一样的拼盘。你随意搭配,你能组出多少种不同的方案,我就买多少份!”

现在就请你帮帮Lele,帮他算一算到底能够卖出多少份水果拼盘给lcy了。

注意,水果是以个为基本单位,不能够再分。对于两种方案,如果各种水果的数目都相同,则认为这两种方案是相同的。

最终Lele拿了这笔钱,又可以继续他的学业了~

输入输出

输入

本题目包含多组测试,请处理到文件结束(EOF)。
每组测试第一行包括两个正整数N和M(含义见题目描述,0<N,M<=100)
接下来有N行水果的信息,每行两个整数A,B(0<=A<=B<=100),表示至少要买该水果A个,至多只能买该水果B个。

输出

对于每组测试,在一行里输出总共能够卖的方案数。
题目数据保证这个答案小于10^9

思路

母函数的应用
对于每种水果 f r u i t i fruit_i fruiti​ 的上下界设为 a i , b i a_i, b_i ai​,bi​ 并设 c i = b i − a i + 1 c_i = b_i-a_i+1 ci​=bi​−ai​+1
则母函数可以写为
∏ i = 1 n ( x a i + x a i + 1 + . . . + a b i ) \prod_{i=1}^n(x^{a_i}+x^{a_i+1}+…+a^{b_i}) ∏i=1n​(xai​+xai​+1+…+abi​)
化简 (等比数列求和) 为
∏ i = 1 n ( x a i 1 − x c i 1 − x ) \prod_{i=1}^n(x^{a_i}\frac{1-x^{c_i}}{1-x}) ∏i=1n​(xai​1−x1−xci​​)
进一步化简为
x ∑ i = 1 n a i × ∏ i = 1 n ( 1 − x c i ) × ∑ i = 1 n ( C n − 1 + i i x i ) x^{\sum_{i=1}^n a^i}\times \prod_{i=1}^n (1-x^{c_i}) \times \sum_{i=1}^n(C_{n-1+i}^ix^i) x∑i=1n​ai×∏i=1n​(1−xci​)×∑i=1n​(Cn−1+ii​xi)
问题要求上式中 x m x^m xm 的系数
上式可以分为三部分, 左边部分直接累加得到, 即
设 t o t a l = ∑ i = 1 n a i total=\sum_{i=1}^na^i total=∑i=1n​ai
那么原问题等价于求
f ( x ) = ∏ i = 1 n ( 1 − x c i ) × ∑ i = 1 n ( C n − 1 + i i x i ) f(x)= \prod_{i=1}^n (1-x^{c_i}) \times \sum_{i=1}^n(C_{n-1+i}^ix^i) f(x)=∏i=1n​(1−xci​)×∑i=1n​(Cn−1+ii​xi) 中
x m − t o t a l x^{m-total} xm−total的系数
对于累乘项 ∏ i = 1 n ( 1 − x c i ) \prod_{i=1}^n (1-x^{c_i}) ∏i=1n​(1−xci​), 直接暴力展开 (因为 m − t o t a l ≤ 100 m-total \le 100 m−total≤100, 所以 最多循环 m × 100 m\times100 m×100 次, 可以接受)
对于累加项 ∑ i = 1 n ( C n − 1 + i i x i ) \sum_{i=1}^n(C_{n-1+i}^ix^i) ∑i=1n​(Cn−1+ii​xi), 由于问题没有要求取模, 所以没法用阶乘来算 (因为最高需要 200 的阶乘), 所以我使用杨辉三角来建表计算.

代码 (C 语言)

  1. #include <stdio.h>
  2. #include <string.h>
  3. long long coeff[101];
  4. long long temp[101];
  5. long long c[100][100];
  6. int m, n;
  7. long long mom(int i) {
  8. return c[n+i-1][i];
  9. }
  10. int main() {
  11. // 杨辉三角建表
  12. for (int i = 0; i <= 100; i++) {
  13. c[i][0] = c[i][i] = 1;
  14. if (i < 2) continue;
  15. for (int j = 1; j < i; j++) c[i][j] = c[i-1][j-1] + c[i-1][j];
  16. }
  17. while (scanf("%d%d", &n, &m) != EOF) {
  18. int total = 0;
  19. memset(coeff, 0, sizeof(coeff));
  20. coeff[0] = 1;
  21. for (int i = 0; i < n; i++) {
  22. int from, to;
  23. scanf("%d%d", &from, &to);
  24. total += from;
  25. int count = to - from + 1;
  26. // 暴力展开累乘项中每一个项的系数
  27. for (int k = 0; k <= 100; k++) temp[k] = coeff[k];
  28. for (int k = 0; k + count <= 100; k++) temp[k+count] += -1 * coeff[k];
  29. for (int k = 0; k <= 100; k++) coeff[k] = temp[k];
  30. }
  31. int need = m - total;
  32. long long ans = 0;
  33. for (int j = 0; j <= need; j++) {
  34. ans += coeff[j] * mom(need-j);
  35. }
  36. printf("%lld\n", ans);
  37. }
  38. }

总结

一个月前的我可能看见母函数就一脸懵逼, 现在总算是会一点了QAQ

发表评论

表情:
评论列表 (有 0 条评论,351人围观)

还没有评论,来说两句吧...

相关阅读

    相关 组合数学选择题

    1.【题目】美团有个传统,就是公司各部门每月都要组织员工进行一次团建互动(team building,简称TB),每个员工都可以带家属参加。活动内容出了吃喝玩之外,还要做一些互