PAT-python-1152 Google Recruitment

水题，大意是第一行输入数字的长度和所求素数的长度，第二行是数字，要求返回数字的子串中第一个长度符合要求的素数子串，没有的话返回404。

直接模拟即可。

测试点2要求返回的是子串而非数字（比如0002要返回0002而非2）。

解法

from math import sqrt

def isp(n):
    if n < 2: return False
    for i in range(2, int(sqrt(n)) + 1):
        if n % i == 0: return False
    return True

n = int(input().split()[-1])
s = input()
for i in range(len(s) - n + 1):
    s1 = s[i:i+n]
    num = int(s1)
    if isp(num):
        print(s1)
        exit(0)
print('404')