PROGRAM MANG;
CONST FI='MANG.INP';
      FO='MANG.OUT';
VAR N,K,I:LONGINT;
    A:ARRAY[1..32000] OF LONGINT;
    F:TEXT;
FUNCTION KT(N:QWORD):BOOLEAN;
VAR I:LONGINT;
    OK:BOOLEAN;
BEGIN
 OK:=FALSE;
 I:=TRUNC(SQRT(N));
 IF SQR(I)=N THEN OK:=TRUE;
 KT:=OK;
END;
BEGIN
 ASSIGN(F,FI);
 RESET(F);
 READLN(F,N,K);
 FOR I:=1 TO N DO READ(F,A[I]);
 CLOSE(F);
 ASSIGN(F,FO);
 REWRITE(F);
 IF KT(A[K]) THEN WRITELN(F,'YES') ELSE WRITELN(F,'NO');
 FOR I:=1 TO N DO
  IF KT(A[I]) THEN WRITELN(F,A[I],' ',I);
 CLOSE(F);
END.

Python:

n = int(input())
d = str(n)
e = list(d)
f = []
for i in e:
      p = int(i)
      f.append(p)
a = len(f) - 1 
if f[a] == 0 or f[a] == 1 or f[a] == 4 or f[a] == 5 or f[a] == 6 or f[a] == 9:
    print(n, "là số chính phương.")
else:
      print(n, "không phải là số chính phương.")