HDOJ--2079选课时间!!!母函数

原题链接
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门。
 

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
 
思路:母函数,给你一些不大于8的数据,试问组成某种学分的方法有几种,母函数这里不在赘述,有兴趣可以看我的母函数的一篇文章——》 母函数思考与讨论  

代码:
#include
#include
int ak[9];			//存每种学分最多的情况 
int c1[650];		//按照题意最大为650,存储最终结果 
int c2[650];		//存储中间结果 
int main()
{
	int T;
	int n,m;							//因为k用到了循环,这里m代替k行 
	int a,b;
	int i,j,k;
	while(scanf("%d",&T)!=EOF)
	{
		while(T--)
		{
			memset(ak,0,sizeof(ak));	//初始化为0 
			scanf("%d%d",&n,&m);
			for(i=1;i<=m;i++)
			{
				scanf("%d%d",&a,&b);
				ak[a]+=b;
			}
			memset(c1,0,sizeof(int)*(n+1));
			memset(c2,0,sizeof(int)*(n+1));
			c1[0]=1;
			for(i=1;i<=8;i++)
			{
				for(j=0;j<=n;j++)
				{
					for(k=0;k<=ak[i]*i&&j+k<=n;k+=i)//只要n学分的组合,最大到n,减树 
					{
						c2[j+k]+=c1[j];
					}
				}
				for(j=0;j<=n;j++)
				{
					c1[j]=c2[j];
					c2[j]=0;
				}
			}
			printf("%d\n",c1[n]);
		}
	}
	return 0;
}


你可能感兴趣的:(算法+ACM)