PAT número automorphic B 1091 N-

El código fue puesto en libertad, aprendemos juntos, ayudarnos unos a otros
Título:

muestra de entrada:

3
92 5 233

Resultado de muestra:

3 25392
1 25
No

Código de abajo (Python):

# PAT 1091
M = int(input())
num_list = map(int, input().split(' '))
result_list = []
for i in num_list:
    flag = True
    for j in range(1, 10):
        temp = i**2 * j
        if temp % (10**len(str(i))) == i:
            result_list.append(str(j)+' '+str(temp))
            flag = False
            break
    if flag:
        result_list.append('No')
for i in result_list:
    print(i)