2832: 第n小的质数

参考代码:

import math
n = int(input())
arr = []
for i in range(2, 100000):
    for j in range(2, int(math.sqrt(i)) + 1):
        if i % j == 0:
            break
    else:
        arr.append(i)
print(arr[n-1])