算法提高 拿糖果 DP

http://lx.lanqiao.cn/problem.page?gpid=T376

注意进行求素数表时，循环的范围。

#include <iostream>
#include <algorithm>
#include <map>
#include <cmath>
#include <queue>
#include <cstring>
using namespace std;
const int maxn=1000005;

int dp[maxn];
int prime[maxn];
int tot=0;
int mk[maxn]={1,1,0};
int n;

void init(){
    for (int i=2;i<=sqrt(n+1);i++){     //如果i<=n会报错
        if (mk[i]) continue;
        prime[tot++]=i;
        for (int j=i;j*i<=n;j++){
            mk[i*j]=1;
        }
    }
}

int main(){

    cin >> n;

    init();

    for (int i=1;i<=n;i++){
        for (int j=0;j<tot;j++){
            if (prime[j]>sqrt(i)) break;
            if (i%prime[j]==0){
                dp[i]=max(dp[i],dp[i-2*prime[j]]+prime[j]);
            }
        }
    }
    cout << dp[n] << endl;
}