蓝桥杯 算法提高-拿糖果(简单dp/记忆化搜索)

题目

N<=1e5

题解

注意不大于根号的条件

所以筛素数的时候，最大的导致break的素因子不放在里面

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
bool ok[maxn];
int prime[maxn],cnt;
int dp[maxn];
int n;
vector<int>res[maxn];
void init() 
{
	memset(dp,-1,sizeof dp);
	for(int i=2;i<maxn;++i)
	{
		if(!ok[i])prime[cnt++]=i;
		for(int j=0;j<cnt;++j)
		{
			int k=i*prime[j];
			if(k>maxn)break;
			ok[k]=1;
			if(i%prime[j]==0)break;
		}
	}
	for(int i=2;i<maxn;++i)
	{
		int tmp=i;
		for(int j=0;j<cnt;++j)
		{
			if(prime[j]*prime[j]>tmp)break;
			if(i%prime[j]==0)
			{
			 res[i].push_back(prime[j]);
			 while(tmp%prime[j]==0)
			 tmp/=prime[j];
		    }
		}
		//if(tmp>1)res[i].push_back(tmp); <=sqrt(m)的搞进去 
	}
}
int dfs(int p)
{
	if(p<0)return 0;
	if(dp[p]!=-1)return dp[p];
	if(p<=3)return dp[p]=0;
	int len=res[p].size(),ans=0;
	for(int i=0;i<len;++i)
	ans=max(ans,dfs(p-2*res[p][i])+res[p][i]);
	return dp[p]=ans;
}
int main()
{
	init();
	scanf("%d",&n);
	printf("%d\n",dfs(n));
	return 0;
}