Cho dãy số nguyên N. Hãy cho biết từ 0 đến N có bao nhiêu số chính phương, bao nhiêu số nguyên tố?

Mấy bạn ơi trả lời giúp mình bằng ngôn ngữ Python 3 ạ. Nhanh giúp mình vs. SOS

import math
def snt(x):
    if x < 2:
        return False
    if x < 4:
        return True
    if x % 2 == 0 or x % 3 == 0:
        return False
    for i in range(5, int(math.sqrt(x)) + 1, 6):
        if x % i == 0 or x % (i + 2) == 0:
            return False
    return True

def cpp(n):
    if n < 0:
        return False
    s = int(math.sqrt(n))
    return s * s == n

def main():
    n = int(input())
    dem = 0
    cp = 0
    for i in range(n + 1):
        k = int(math.sqrt(i))
        if snt(i):
            dem+=1
        if cpp(i):
            cp+=1
    print(cp, dem)
main()

def snt(n):
    if n<2: return False
    k=int(n**0.5)
    for i in range(2,k+1):
        if n%i==0: return False
    return True
n=int(input())
s=1
x=0
for i range (2,n+1):
    #meothinhle
    k=int(i**0.5)
    if k*k==i:
        s+=1
    if snt(i):
        x+=1
print(f"Số chính phương từ 0 đến N là có {s} số")
print(f"Số số nguyên tố từ 0 đến N là có {x} số")