华为机试——自守数

题目：自守数

自守数是指一个数的平方的尾数等于该书自身的自然数。例如25^2=625,76^2=5776,9376^2=87909376。

求出n以内的自守数的个数

#include<iostream>
using namespace std;

int CalAutomorphicNumbers(int n)
{
    int count=0;
    int i;
    for(i=1;i<=n;i++)
    {
        long num=i*i;
        if(!(i/10))
        {
            if(num%10 == i)
                count++;
        }
        else if(!(i/100))
        {
            if(num%100==i)
                count++;
        }
        else if(!(i/1000))
        {
            if(num%1000 == i)
                count++;
        }
        else if(!(i/10000))
        {
            if(num%10000==i)
                count++;
        }
        else if(!(i/100000))
        {
            if(num%100000==i)
                count++;
        }
    }
    return count;
}

int main()
{
    int num;
    cin>>num;
    cout<<CalAutomorphicNumbers(num)<<endl;
    return 0;
}