华为OJ-自守数

 1 #include <iostream>
 2 using namespace std;
 3 int CalcAutomorphicNumbers( int n)
 4 {
 5 
 6     int count=0;
 7     for(int i=0;i<=n;i++)
 8     {
 9         int  sum=i*i;
10         if (!(i/10))   //一位数
11         {
12             if(sum%10==i)
13                 count++;
14         }
15         else if (!(i/100))   //两位数
16         {
17             if(sum%100==i)    
18                 count++;
19         }
20         else if (!(i/1000))   //三位数
21         {
22             if(sum%1000==i)    
23                 count++;
24         }
25         else if (!(i/10000))   //四位数
26         {
27             if(sum%10000==i)
28                 count++;
29         }
30     }
31 
32     return count;
33 }
34 
35 int main()
36 {
37     int num;
38     cin>>num;
39     cout<<CalcAutomorphicNumbers(num);
40 }