问题 A: 守形数

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

6
11

样例输出

Yes!
No!

#include<iostream>
using namespace std;
int main() {
	int num,a;
	int i, j;
	
	while (cin >> num) {
		a = num * num;
		int flag = 1;
		while (num != 0) {
			i = num % 10;
			j = a % 10;
			if (i != j) {
				cout << "No!" << endl;
				flag = 0;
				break;
			}
			num /= 10;
			a /= 10;
		}
		if (flag) {
			cout << "Yes!" << endl;
		}
	}
	return 0;
}