http://acm.hdu.edu.cn/showproblem.php?pid=2189
灾区来了n位志愿者,抗震救灾指挥部需要将他们分为若干个小组,小组的数量不限,但是要求每个小组的人数必须为素数,请问我们有几种分组的方法呢?
input
输入数据首先包含一个正整数C,表示有C组测试用例,然后是C行数据,每行包含一个正整数n(2<=n<=150),表示志愿者的总人数。
output
对于每组测试数据,请输出分组的方案数目,每个输出占一行。
sample input
3
3
4
5
sample output
1
1
2
思路一:搜索,写完程序后发现,算到150需要1200ms左右的时间,题目要求1000ms以内,试着提交了一下,果然Time out
思路二:贪心,把每次的结果保存下来,后面的利用前面的来做,发现数据怎么都对不上
思路三:母函数,发现此题很符合母函数的思路,每个素数都可以取无限个,于是摸索了一便母函数的思路,AC!
//代码
#include <stdio.h>
#define lmax 155
int c1[lmax+1],c2[lmax+1];
int s[lmax+1];
int main()
{
int t,n,i,j,k;
for(i=0;i<lmax+1;i++)
s[i]=1;
s[0]=s[1]=0;
for(i=2;i<lmax+1;i++)//初始化素数
for(j=2;j<(lmax+1)/i;j++)
s[i*j]=0;
scanf("%d",&t);
while (t--)
{
scanf("%d",&n);
for (i=0;i<=n;i++)
{c1[i]=0; c2[i]=0;}
for (i=0;i<=n;i++) if( i%2==0 )c1[i]=1;
for (i=3;i<=n;i++)
{
if(s[i]){
for (j=0;j<=n;j++){
for (k=0;k+j<=n;k+=i)
{
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;
}