UVA1213Sum of Different Primes(素数打表 + DP)

题目链接

题意：选择k个素数，使得和为N（1120）的方案数；

筛选出 <= N 的素数，然后就背包

写的时候没初始dp[0][0] = 1;而且方案数也没相加，真是弱逼

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int Max = ;
int prime[Max + ], total,flag[Max + ];
int dp[Max + ][];
void get_prime()
{
    total = ;
    memset(flag, , sizeof(flag));
    for(int i = ; i <= Max; i++)
    {
        if(flag[i] == )
        {
            prime[ ++total ] = i;
            for(int j = i; j <= Max / i; j++)
                flag[j * i] = ;
        }
    }
}
int main()
{
    get_prime();
    int n,k;
    while(scanf("%d%d", &n, &k) != EOF)
    {
        if(n ==  && k == )
            break;
        memset(dp, , sizeof(dp));
        dp[][] = ;  //初始
        for(int i = ; i <= total; i++)  //对于每一个素数进行判断
        {
            for(int j = n; j >= prime[i]; j--)  //对于每一个能选择素数的判断
            {
                for(int m = ; m <= k; m++)  //选择的个数可以使1 ... k
                {
                    if(dp[j - prime[i]][m - ] != )
                        dp[j][m] += dp[j - prime[i]][m - ];  //当前状态选择m个素数和为j  == 选择m - 1个素数合为 j - prime[i]
                }
            }
        }
        printf("%d\n", dp[n][k]);
    }
    return ;
}