Задача "Т-простые числа"

from math import sqrt
a = int(input())
b = list(map(int, input().split()))
 
for i in range(a):
    c = sqrt(b[i])
    cc = int(sqrt(c)) + 1
    if c in [2, 3]:
        print('YES')
    elif b[i] == 1:
        print('NO')
    elif b[i] % 2 != 0:
        if b[i] % 3 != 0:
            if (c + 1) % 6 == 0 or (c - 1) % 6 == 0:
                for s in range(5, cc, 2):
                    if c % s == 0:
                        print('NO')
                        break
                else:
                    print('YES')
            else:
                print('NO')
        else:
            print('NO')
    else:
        print('NO')

def is_t_prime(n):   
    divisors_count = 0
    for i in range(1, int(n**0.5) + 1):
        if n % i == 0:
            if i * i == n: # Если делитель - корень, считаем один раз
                divisors_count += 1
            else: # Иначе считаем два делителя (i и n/i)
                divisors_count += 2
        if divisors_count > 3: 
            break
    return 'YES' if divisors_count == 3 else 'NO'
 
a = int(input()) # вроде как лишняя строка но можно использовать как индексы (ваш вариант)
for b in list(map(int, input().split())):
    print(is_t_prime(b))

math.isqrt(x)**2==x

n = 1000000
primes = [False] * 2 + [True] * (n - 1)
i = 2
while i * i <= n:
    if primes[i]:
        for j in range(i * i, n + 1, i):
            primes[j] = False
    i += 1
    
input()
for x in map(int, input().split()):
    sqrt_x = int(x ** 0.5)
    print('YES' if sqrt_x * sqrt_x == x and primes[sqrt_x] else 'NO')

import math
 
l = [int(input('Число ')) for i in range(int(input('Количество ')))]
res = []
 
def find_divisors(n):
    global res
    divisors = []
    for i in range(1, int(math.sqrt(n)) + 1):
        if not n % i:
            divisors.append(i)
            if i != n // i:
                divisors.append(n // i)
    return divisors
 
for i in range(len(l)):
    res += [find_divisors(l[i])]
result = [l[i] for i in range(len(l)) if len(res[i]) == 3]
for i in l:
    if i in result:
        print('YES', i)
    else:
        print('NO', i)

import math
import time
 
start_time = time.time()
 
l = list(map(int,'6 9 8 7 25'.split(' ')))
res = []
 
def find_divisors(n):    
    divisors = []
    for i in range(1, int(math.sqrt(n)) + 1):
        if not n % i:
            divisors.append(i)
            if i != n // i:
                divisors.append(n // i)
    return divisors
 
 
for i in range(len(l)):
    res += [find_divisors(l[i])]
result = [l[i] for i in range(len(l)) if len(res[i]) == 3]
for i in l:
    if i in result:
        print('YES', i)
    else:
        print('NO', i)
        
end_time = time.time() 
execution_time = end_time - start_time 
 
print(f"Время выполнения программы: {execution_time} секунд")

l = list(map(int,input().split(' ')))
start_time = time.time()