PAT nombre automorphic B 1091 N-

Le code a été libéré, nous apprenons ensemble, entraidons
Titre:

Exemple d' entrée:

3
92 5 233

Exemple de sortie:

3 25392
1 25
No

Code ci-dessous (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)