4个质数和
备用链接
——————————————————————————————————————————————
一眼看去,想到dp
很明显是经典的背包问题,就直接打了。。。
——————————————————————————————————————————————
70分做法:
其实我是当满分做法做的,但时间复杂度分析错了,
时间复杂度应是
O
(
N
k
)
O(Nk)
O(Nk),N表示输入的最大的数,k表示N以下有多少个质数,我想当然地将k估算成
N
\sqrt{N}
N,但进过打表后,发现最大数据N=100000时,k=9592,远大于
N
\sqrt{N}
N,于是就T了,
套用经典的背包问题,再降维之后,列出
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j]表示乘积为i,用了j个质数的方案总数
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100100;
int cnt=0,n;
int prime[N];
ll dp[N][5];
bool vis[N];
void get_prime(){
for(int i=2;i<=100000;i++)
{
if(!vis[i]){ prime[++cnt]=i;}
for(int j=1;j<=cnt&&prime[j]*i<=100000;j++){
vis[prime[j]*i]=1;
if(i%prime[j]==0){
break;
}
}
}
}
int MAX=1;
int main(){
int T;
scanf("%d",&T);
get_prime();
for(int i=1;i<=n;i++)dp[i][1]=0,dp[i][2]=0,dp[i][3]=0,dp[i][4]=0;
for(int i=1;i<=cnt;i++) dp[prime[i]][1]=1;
while(T--){
scanf("%d",&n);
if(n<MAX){
printf("%lld\n",dp[n][4]);
continue;
}
for(int i=MAX+1;i<=n;i++)
for(int j=1;j<=cnt&&prime[j]<=n;j++)
for(int k=1;k<=4;k++){
if(prime[j]<=i){
dp[i][k]+=dp[i-prime[j]][k-1];
}
}
printf("%lld\n",dp[n][4]);
MAX=max(MAX,n);
}
}
——————————————————————————————————————————————
100分做法:
考虑优化,
可
以
将
d
p
[
i
]
固
定
为
两
个
质
数
,
和
为
i
的
方
案
总
数
可以将dp[i]固定为两个质数,和为i的方案总数
可以将dp[i]固定为两个质数,和为i的方案总数
这样时间就优化到了
O
(
K
2
)
O({K^2})
O(K2)
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100010;
int n,prime[10000],cnt=0;
ll dp[N];
bool vis[N];
void get_prime()
{
for(int i=2;i<=100000;i++){
if(!vis[i]){
prime[++cnt]=i;
}
for(int j=1;j<=cnt&&prime[j]*i<=100000;j++){
vis[prime[j]*i]=1;
if(i%prime[j]==0) break;
}
}
}
int main(){
int T;
scanf("%d",&T);
get_prime();
for(int i=1;i<=cnt;i++)
for(int j=1;j<=cnt&&(prime[j]+prime[i])<=100000;j++){
dp[prime[j]+prime[i]]++;
}
while(T--){
scanf("%d",&n);
ll ans=0;
for(int i=0;i<=n;i++){ans+=1ll*dp[i]*dp[n-i];}
printf("%lld\n",ans);
}
}
这道题说明了时间复杂度如果是单调递增的,就直接代入最大值检验