可优化-PAT (Basic Level) Practice Python解法 1094 谷歌的招聘（素数判定）

可优化-PAT (Basic Level) Practice Python解法 1094 谷歌的招聘（素数判定）

import math

def isprime(n):
    n = int(n)
    if n==1 or (n!=2 and n%2 == 0):
        return False
    if n==2 or n==3:
        return True
    
    prime = True
    for k in range(3,int(math.sqrt(n)+1),2):
        if n%k==0:
            prime = False
            break
    return prime

num, long = map(int,input().split(' '))
sample = input()
for i in range(num-long+1):
    ck = sample[i:i+long]
    if isprime(int(ck)):
        print(ck)
        break
else:
    print('404')

实际上素数运算的判定里，把已知的素数加到除数里可以大量减少判定时间，但是我不知道怎么在一重def下进行这个运算。
常见的素数判定方法（方法2的最后一个坑点会报错，有特殊值）素数判断算法（基于python实现）