【C++】马蹄集02 自守数II

输入正整数N，检查该数是否为自守数输出YES或者NO。当且仅当一个数的平方以与该数相同的数字结尾时该数称为自守数。 

格式

输入格式：输入正整数N
输出格式：输出YES或者NO

 样例1

输入：76
输出：YES

代码  利用substr

#include<bits/stdc++.h> 
#include<string>

using namespace std;

int main( )
{
    int n,n2;
    string s,s2;
    cin>>n;
    n2=n*n;
    s=to_string(n);
    s2=to_string(n2);
    if(s2.substr(s2.size()-s.size())==s) cout<<"YES";
    else cout<<"NO";
    return 0;
}