HDU-2079 选课时间(题目已修改,注意读题) -母函数
选课时间(题目已修改,注意读题)
Time Limit: 1000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 995 Accepted Submission(s): 836
Problem Description
又到了选课的时间了,xhd看着选课表发呆,为了想让下一学期好过点,他想知道学n个学分共有多少组合。你来帮帮他吧。(xhd认为一样学分的课没区别)
Input
输入数据的第一行是一个数据T,表示有T组数据。
每组数据的第一行是两个整数n(1 <= n <= 40),k(1 <= k <= 8)。
接着有k行,每行有两个整数a(1 <= a <= 8),b(1 <= b <= 10),表示学分为a的课有b门。
每组数据的第一行是两个整数n(1 <= n <= 40),k(1 <= k <= 8)。
接着有k行,每行有两个整数a(1 <= a <= 8),b(1 <= b <= 10),表示学分为a的课有b门。
Output
对于每组输入数据,输出一个整数,表示学n个学分的组合数。
Sample Input
2 2 2 1 2 2 1 40 8 1 1 2 2 3 2 4 2 5 8 6 9 7 6 8 8
Sample Output
2 445
简单的母函数在边界上选取 N 更好 ,前面数组越界,程序诡异的要死啊。
#include<stdio.h> #include<stdlib.h> #include<math.h> #include<string.h> #include<time.h> int n1[45],n2[45],sub[10],N,K2; void generating_function( ) { memset(n1,0,sizeof(n1)); for(int i=0;i<=sub[1];++i) n1[i]=1; memset(n2,0,sizeof(n2)); for(int i=2;i<=K2;++i) { for(int j=0;j<=N;++j) for(int k=0;k<=sub[i]&&j+k*i<=N;++k) { n2[j+k*i]+=n1[j]; } for(int u=0;u<=N;++u) { n1[u]=n2[u]; n2[u]=0; } } } int main() { int T; scanf("%d",&T); while(T--) { scanf("%d%d",&N,&K2); for(int i=1;i<=K2;++i) { int i; scanf("%d",&i); scanf("%d",&sub[i]); } generating_function(); printf("%d\n",n1[N]); } return 0; }