比较偏向于暴力,不一定是正解,时间再 2s 多一点,最慢的一个跑到 800ms 左右,能过就好。
我们确定数组的话,应该是找到最大的可以选择的数的下一个数,所以先把判断是否符合规定的函数写好:
inline bool p(int k){
while(k){
if(k%10==7)
return 0;
k/=10;
}
return 1;
}
由这个东西可以得知,最后的数应该是 ,如果最大的要查找的 x 不是 ,在它的两倍的范围内一定有。然后我们数组在这个基础上稍微开大点就行。然后我们可以找到一个不符合规定的数就把它的倍数删掉。但为了避免 7 的倍数重复出现,我们可以特判删除。答案用 nxt 存储,前面的不符合规定的数赋值为 -1。查找符合规定的下一个数可以用搜索,初始化从 0 开始,然后再找到下一个符合规定的数 k,若当前查找的为 i,那么 即可。
这里还有一个玄学方法,虽然复杂度一样,但也许更慢:由“唯一分解定理”可知,如果一个含 7 的数字 k,那么我们只需要删除它的倍数即可。理由: ( 为素数, 为正整数) 是一定可以被删除的,因为它可以这样分解。但是这个东西似乎没有效果,还会让您的程序跑得更慢,但至少能过。(我赛时把 写成 似乎要死)。
时间复杂度根据数据来定,应该远小于 ,似乎可以看成 。
#include<bits/stdc++.h>
#define min(a,b) a<b?a:b
using namespace std;
const int maxk=1e7+5,maxf=2e5+5;
int T,x[maxf],ans[maxk],nxt[maxk],cnt,now,maxn,k;
bool prime[maxk];
inline void Prime(int n){
for(int i=2;i<=n;++i){
if(!prime[i])
ans[++cnt]=i;
for(int j=1;j<=cnt&&ans[j]*i<=n;++j){
prime[ans[j]*i]=0;
if(!(i%ans[j]))
break;
}
}
}
inline bool p(int k){
while(k){
if(k%10==7)
return 0;
k/=10;
}
return 1;
}
inline void init(int now){
while(now<=maxn){
for(int i=now+1;;++i){
if(!(i%7)){
nxt[i]=-1;
} else if(nxt[i]==-1){
for(int j=1;j<=cnt&&i*ans[j]<=k;j++)
nxt[ans[j]*i]=-1;
} else if(p(i)){
nxt[now]=i;
now=i;
break;
} else{
nxt[i]=-1;
for(int j=1;j<=cnt&&i*ans[j]<=k;j++)
nxt[ans[j]*i]=-1;
}
}
}
}
int main(){
scanf("%d",&T);
for(int i=1;i<=T;++i)
scanf("%d",&x[i]),maxn=max(maxn,x[i]);
k=min(1e7+1,maxn*2);
Prime(k);
init(0);
for(int i=1;i<=T;++i)
printf("%d\n",nxt[x[i]]);
return 0;
}