
t의 갯수를 받아주고 갯수만큼 반복문을 돌려준다음, 입력받은 n을 곱해서 square에 저장해준다. 그다음 str(n)의 길이를 구하여 저장을 해주면 만약 n_square에서 -N_length: 한값이 str(N)과 같으면 yes를 출력해주고 아니면 no를 출력해주면 되는 간단한 문제이다.
# 2028
T = int(input())
for test_case in range(T):
N = int(input())
N_square = str(N * N)
N_length = len(str(N))
if N_square[-N_length:] == str(N):
print("YES")
else:
print("NO")
https://www.acmicpc.net/problem/2028