守形数

题目描述
守形数是这样一种整数，它的平方的低位部分等于它本身。 比如25的平方是625，低位部分是25，因此25是一个守形数。 编一个程序，判断N是否为守形数。
输入描述:
输入包括1个整数N，2<=N<100。
输出描述:
可能有多组测试数据，对于每组数据，
输出"Yes!”表示N是守形数。
输出"No!”表示N不是守形数。
示例1
输入
25
4
输出
Yes!
No!

#include <stdio.h>
#include <iostream>
#include <algorithm>
using namespace std;
int main(){
    int N;
    int M;
    while(scanf("%d",&N)!=EOF){
        M = N*N;
        if(N<10){
            int t = M%10;
            if(t == N){
                printf("Yes!\n");
            }
            else
                printf("No!\n");
        }
        else if(N>=10)
        {
            int t = M%100;
            if(t == N){
                printf("Yes!\n");
            }
            else
                printf("No!\n");
        }
    }
    return 0;
}