谷歌的招聘
import math
def isprime(a):
up = int(math.sqrt(a)+1)
if a == 2 or a == 3:
return True
if a == 1 or a == 0:
return False
if a%6!=1 and a%6!=5:
return False
for i in range(5,up,6):
if a%i==0 or a%(i+2)==0:
return False
return True
k,p = [int(i) for i in input().split()]
n = input()
flag = 0
for i in range(p-1,len(n)):
if int(n[i])%2==0 and int(n[i-p+1:i+1])!=2:
pass
else:
j = int(n[i-p+1:i+1])
if isprime(j):
print(str(j).zfill(p))
flag = 1
break
if flag == 0:
print('404')
啊啊啊啊啊这里要给大家分享一个我刚刚在百度上发现的一个判断素数的超快算法!
https://blog.csdn.net/huang_miao_xin/article/details/51331710 ←戳链接哈哈哈哈
大概就是大于5的素数全部都在6的倍数的两侧,证明在链接里。然后判断的时候,也是隔着6个6个判断就可以了。感觉超快的。
话说回来,这题我的方法比如样例是 20 5 那就从第五个字符串开始判断,如果是偶数就直接跳过,因为不可能是偶数(但是就是因为这个,我做的过程中也漏下了00002这种可能),如果是奇数,就拿这个字符和她前面四个字符拼在一起去判断,用我发现的那个判断素数的超快办法去做。然后就okk啦。