首页 > 代码库 > 找单词(母函数)
找单词(母函数)
例题题:找单词
母函数的介绍
母函数百度百科
假设数据给出A、B、C、D(价值:1、2、3、4),个数分别为a[1],a[2],a[3],a[4].
生成函数(母函数)为:(x^0 + x^1 + x^2 + ... + x^a[1] ) * ( x^0 + x^2 + x^4 + ... + x^2*a[i] ) * (x^0 + x^3 + x^6 + ... + x^3*a[]i]) * (x^0 + x^4 + x^8 + ... + x^4*a[4])
对应数列c[0],c[1],c[2],c[3],c[4]分别为x^0,x^1,x^2,x^3,x^4的系数(对应“找单词”这道题对应系数即为对应价值的组合个数c[1]大小就是价值为1的组合个数)
求c序列的思路:此处易知x^1的系数为1,x^2的系数为1,x^3的系数为x^3的系数加上x^1的系数乘以x^2的系数,x^4的系数为x^4的系数加上x^1的系数乘以x^3再加上x^2系数的平方(即x^2的系数乘以x^2的系数)......
具体实现见代码:
#include<cstdio> #include<cstring> const int maxn = 50; int c1[maxn + 1], c2[maxn + 1], a[30]; int main(){ int T, n; scanf("%d", &T); while(T--){ memset(c1, 0, sizeof(c1)); c1[0] = 1; memset(c2, 0, sizeof(c2)); for(int i = 1; i <= 26; i++) scanf("%d", &a[i]); /******************************************** *************母函数的三层嵌套循环************* ********************************************* ******第一层循环是总的种类个数(26个字母)***** ********************************************* ******第二层循环是每个种类的数目(a[i])******* ********************************************* **第三层循环的计数变量是每项系数(答案)的大小** c1的下标即从0到maxn-j*i ********************************************/ for(int i = 1; i <= 26; i++){ for(int j = 0; j <= a[i]; j++) for(int k = j*i; k <= maxn; k++) c2[k] += c1[k - j*i]; memcpy(c1, c2, sizeof(c2)); memset(c2, 0, sizeof(c2)); } n = 0; for(int i = 1; i <= maxn; i++) n += c1[i]; printf("%d\n", n); } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。