判断守形数

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

int fun(int x)
{
	int tmp, res = 0,e=1,t=0;
	while (x > 9)
	{
		tmp = x % 10;
		x = x / 10;
		res += tmp*e;
		e = 10 * e;
		++t;
		if (t >= 2)break;
	}
	printf("%d\n", res);
	return res;
}

void solve()
{
	int x,dw,pf;
	while (~scanf("%d", &x))
	{
		pf = x*x;
		dw = fun(pf);
		if (dw == x)printf("YES\n");
		else printf("NO\n");
	}
}