POJ 3132 Sum of Different Primes

题目链接：http://poj.org/problem?id=3132

题意：给定n，k，求选k个小于等于n的素数之和为n的方案数。

思路：范围内有100多个素数，搜索肯定是不行的，所以我们考虑用dp来解决。f[i][j]表示选i个素数，和为j的方案数。那么对于当前的数x，就有f[i][j] += f[i-1][j-x]，然后去枚举i和j。f[0][0] = 1，f[k][n]即为答案。

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <stack>
#include <map>
#include <set>
#include <vector>
#include <sstream>
#include <queue>
#include <utility>

using namespace std;

#define rep(i,j,k) for (int i=j;i<=k;i++)
#define Rrep(i,j,k) for (int i=j;i>=k;i--)

#define Clean(x,y) memset(x,y,sizeof(x))
#define LL long long
#define ULL unsigned long long
#define inf 0x7fffffff
#define mod %100000007

int n,k;

int f[15][1200];
int ans;

int prime[200];
int num;
bool isprime(int x)
{
    for(int i = 2; i * i <= x; i++)
        if ( x % i == 0 ) return false;
    return true;
}
void init()
{
    num = 0;
    rep(i,2,1120)
        if ( isprime(i) )
            prime[++num] = i;
}

int main()
{
    init();//先将素数处理出来
    while(~scanf("%d%d",&n,&k))
    {
        if ( n + k == 0 ) break;
        int len = upper_bound(prime+1,prime+num+1,n) - prime - 1;
        Clean(f,0);
        f[0][0] = 1;
        ans = 0;
        rep(i,1,len)
            Rrep(j,n,prime[i])
            rep(p,1,k)
            {
                f[p][j] += f[p-1][j-prime[i]];
            }
        printf("%d\n",f[k][n]);
    }
    return 0;
}