Peng Lv

毋意,毋必,毋固,毋我。 言必行,行必果。

导航

POJ 3132 Sum of Different Primes (DP)

题意:给定n和k,要求找出k个互不相同的素数,使其和==n,求这样的组合有多少中

思路:因为素数互不相同,自然想到了0-1背包,可是怎么背包却想了很久,因为多了一个限制:k,那么必须用两维数组来做,用dp[M][K]表示在组成M时,K个素数互不相同的情况,于是可以用0-1背包来对每个prime进行dp:

  dp[m][k] = dp[m][k] + dp[m-prime[i]][k-1]; //注意数组不要越界

写完后发现代码运行需要500ms,别人的都是1xms,怎么差别这么大?看了别人的思路和自己的差不多,可是问什么时间差这么多呢?希望路过的神牛指点一二。

#include <iostream>
#include
<cstdio>
#include
<algorithm>
#include
<memory.h>
#include
<cmath>
#include
<set>
#include
<bitset>
#include
<vector>
using namespace std;

const int BORDER = (1<<20)-1;
const int MAXSIZE = 37;
const int MAXN = 1250;
const int INF = 0x7ffffff;
#define CLR(x,y) memset(x,y,sizeof(x))

int n,ans,k,index;
int prime[MAXN],arr_dp[MAXN][MAXN];
int init()
{
CLR(arr_dp,
0);
arr_dp[
0][0] = 1;
return 0;
}
bool _prime(const int& num)
{
int tmp = (int)sqrt(float(num))+1;
for(int i = 2; i < tmp; ++i)
if(!(num%i))
return false;
return true;
}
int find_prime()
{
int i,j;
index
= 0;
for(i = 2; i <= 1120; ++i)
if(_prime(i))
prime[index
++] = i;
return 0;
}
int dp()
{
int bor = index;
for(int i = 0; i < index; ++i)
if(prime[i] > n)
{
bor
= i;
break;
}
for( int i = 0; i < bor; ++i)
for(int j = n; j >= prime[i]; --j)
for(int h = k; h >= 1; --h)
arr_dp[j][h]
+= arr_dp[j-prime[i]][h-1];
OUT(arr_dp[n][k]);
return 0;
}
int main()
{
int i,j,tmp;
find_prime();
while(scanf("%d%d",&n,&k))
{
if(!n && !k)
break;
init();
dp();
}
return 0;
}

 

posted on 2010-04-10 09:36  Lvpengms  阅读(658)  评论(1编辑  收藏  举报