问题 A: 守形数

题目链接：http://codeup.cn/problem.php?cid=100000588&pid=0

题目描述
守形数是这样一种整数，它的平方的低位部分等于它本身。
比如25的平方是625，低位部分是25，因此25是一个守形数。
编一个程序，判断N是否为守形数。

输入
输入包括1个整数N，2<=N<100。

输出
可能有多组测试数据，对于每组数据，
输出"Yes!”表示N是守形数。
输出"No!”表示N不是守形数。

样例输入
6
11

样例输出
Yes!
No!

代码

#include<stdio.h>
#include<string.h>

int to_array(int n, int num[]) {
	int i;
	for(i = 0; n >= 1; i++) { 
		num[i] = n % 10;
		n /= 10;
	}
	return i;
}

int main() {
	int n, nn, i, num_n[3], num_nn[10];
	while(scanf("%d", &n) != EOF)  {
		int len = to_array(n, num_n);
		nn = n * n;
		to_array(nn, num_nn);
		for(i = 0; i < len; i++) {
			if(num_n[i] != num_nn[i]){
				printf("No!\n");
				break;
			}
		}
		if(i == len)
			printf("Yes!\n");
	}
	return 0;
}