题解 | 守形数

#include <bits/stdc++.h>
using namespace std;

int main(){
    int n;
    while(cin>>n){
        int k=n*n,flag=0;
        if(n<10){
            k%=10;
            if(k==n)flag=1;
        }else {
            k%=100;
            if(k==n)flag=1;
        }
        if(flag==1)cout<<"Yes!"<<endl;
        else cout<<"No!"<<endl;
    }
}

这个判断逻辑基于题目逻辑