#include<bits/stdc++.h>
using namespace std;
long long n,s; 
int main(){
	
	cin >> n;
	s=n*n;
	int k=n,i=1;
	while(k!=0){
		k = k/10;
		i = i*10;
	}
	if((s-n)%i==0){
		cout << "YES";
	}
	else{
		cout << "NO";
	}
return 0;
}

C++ :

#include <bits/stdc++.h>
using namespace std;
long long n,s; 
int main(){
	
	cin >> n;
	s=n*n;
	int k=n,i=1;
	while(k!=0){
		k = k/10;
		i = i*10;
	}
	if((s-n)%i==0){
		cout << "YES";
	}
	else{
		cout << "NO";
	}
	return 0;

} 

Python :

# coding=utf-8
n=int(input())
k=n*n
f=1
while(n>0):
    if(k%10!=n%10):
        f=0
    k=k//10
    n=n//10
if(f==0):
    print('NO')
else:
    print('YES')