算法提高 拿糖果 【完全背包】

问题描述

　　妈妈给小B买了N块糖！但是她不允许小B直接吃掉。
　　假设当前有M块糖，小B每次可以拿P块糖，其中P是M的一个不大于根号下M的质因数。这时，妈妈就会在小B拿了P块糖以后再从糖堆里拿走P块糖。然后小B就可以接着拿糖。
　　现在小B希望知道最多可以拿多少糖。

输入格式

　　一个整数N

输出格式

　　最多可以拿多少糖

样例输入

15

样例输出

6

数据规模和约定

　　N <= 100000

思路：

完全背包问题。可以将糖果总数看做背包容量，质因子看做物品。先用素数筛筛出所有素数。

dp[i]=max(dp[i], dp[i-2*prime[j]]+prime[j]);

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<stdlib.h>
#include<math.h>
using namespace std;
int n,fac[100000],fn;
int dp[100005];
#define MAX_N 100005
int prime[MAX_N/2];
bool is_prime[MAX_N+1];
void sieve(int n)
{
    int p = 0;
    memset(is_prime,true,sizeof(is_prime));
    is_prime[0] = is_prime[1] = false;
    for(int i = 2; i <= n/2; i++)
    {
        if(is_prime[i])
            prime[p++] = i;
        for(int j=0; j<p && prime[j]*i<=n; j++)
        {
            is_prime[prime[j]*i]=false;
            if(i%prime[j]==0) 
                break;
        }
    }
}

int main()
{
    memset(dp,0,sizeof dp);
    scanf("%d",&n);
    sieve(n+5);
    for(int i=4;i<=n;i++)
    {
        for(int j=0;prime[j]*prime[j]<=i;j++)
        {
            if(i%prime[j]==0)
            dp[i]=max(dp[i],dp[i-2*prime[j]]+prime[j]);
        }
    }
    printf("%d\n",dp[n]);
    return 0;
}