水果篮子(母函数)
题目:
Description
西瓜的表弟小西瓜生病住院了,西瓜想去买一个水果篮子探望他。水果店里面有很多种类的水果篮子,价格相同,但是水果的搭配各不相同。西瓜突然想到了一个问题,现在水果店里面有这么N种水果,第i个水果单价是Pi元,西瓜手上有M元钱(钱不一定要花完,但也不能什么水果都没有),一共有几种搭配水果篮子的方法呢。
Input
题目包含多组输入,EOF结束,数据最多不超过100组,对于每组数据,包含两行,第一行是两个整数N,M,表示水果的总数和西瓜手里的钱数。第二行包含N个整数,表示每种水果的单价。
1 <= N <= 10, 1 <= M <= 200, 1 <= Pi <= M
1 <= N <= 10, 1 <= M <= 200, 1 <= Pi <= M
Output
对于每组输入,输出一行,表示有多少种搭配水果篮子的方法。
Sample Input
Original | Transformed |
2 10 3 4 9 100 5 6 9 13 4 5 3 9 8
Sample Output
Original | Transformed |
7 1954041
分析:母函数,注意c1,c2数组的初始化
代码:
#include<iostream> #include<cstdio> #include<queue> #include<string> #include<cstring> #include <cmath> using namespace std; int num[11]; int c1[205], c2[205]; int main() { int n,m; while (scanf("%d%d", &n, &m) != EOF) { int sum = 0; for (int i = 1; i <= n; i++) scanf("%d", &num[i]); for (int i = 0; i <= m; i++) { if (i%num[1] == 0) c1[i] = 1; else c1[i] = 0; c2[i] = 0; } for (int i = 2; i <= n; i++) { for (int j = 0; j <= m; j++) for (int k = 0; k + j <= m; k += num[i]) c2[k + j] += c1[j]; for (int j = 0; j <= m; j++) { c1[j] = c2[j]; c2[j] = 0; } } for (int i = 1; i <= m; i++) if (c1[i] != 0) sum+=c1[i]; printf("%d\n", sum); } return 0; }