foj1695 队员选拔问题

Problem Description

• 寒假期间，对参加ACM有兴趣的同学们纷纷练习了很多题目。开学后，他们把自己做题的题目数量和题号发给ths。
  （以下内容纯属虚构）可由于人员太多，而参加名额有限，daxia就提出了一个计算成绩的方法：
  一个同学完成的n道题中，选择任意数量的题号相加，从而能得到的不同素数的个数，就是这个同学的做题成绩。
  现在，让给你写一个程序来帮忙计算做题的成绩。

Input

• 输入有多组数据，每组数据有两行。第一行有一个整数n（不超过50），表示做题数。第二行有n个整数，表示题号（不超过1000，保证不重复）。最后以一个负数表示输入结束，不需要处理。

Output

• 对每一组数据输出一行结果，输出上述的做题成绩。

Sample Input
2
1 2
-1
Sample Output
2

思路：打一个50000以内的素数表。通过dp数组把n个数所有的可能和求出来。如果是素数就使结果加一。

代码：

#include<iostream>
#include<cstdio>
using namespace std;

int prime[50005]={0};

int main()
{
   //打素数表
    for(int i=2;i<=50000;i++)
    {
        if(prime[i]==0)
        for(int j=2*i;j<=50000;j+=i)
            prime[j]=1;
    }
    prime[0]=1;
    int n;
    while(scanf("%d",&n)&&n>=0)
    {
        int a[55],sum=0;
        for(int i=1;i<=n;i++)
            {
                scanf("%d",&a[i]);
                sum+=a[i];
        }
        int dp[50005]={0};
        dp[0]=1;
        //求n个数可能的和
        for(int i=1;i<=n;i++)
        {
            for(int j=sum-a[i];j>=0;j--)
            {
                if(dp[j]==1)
                    dp[j+a[i]]=1;

            }
        }

        int res=0;
        for(int i=2;i<=sum;i++)
        {
            if(dp[i]&&prime[i]==0)
                res++;
        }
        printf("%d\n",res);
    }
    return 0;
}