pta 7-218 自守数

数学的世界有很多很漂亮的数字或者数字集合，比如梅森数，哥德巴赫猜想，水仙花数，完全数，自守数。今天我们就一起来探究一下自守数。自守数是指一个数的平方的尾数等于该数自身的自然数。例如：25^2 = 625，76^2 = 5776，9376^2 = 87909376。请求出n以内的自守数的个数

输入要求

第一行输入n,随后输入n个数m。

输出要求

在一行中输出m以内自守数的个数，所有输出在一行内完成，末尾没有多余空格。

输入示例

5
1
2
3
4
5

输出示例

2 2 2 2 3

---

代码参考

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int n,max=0;
    cin>>n;
    int search[n];
    for(int i=0;i<n;i++){
        cin>>search[i];
        if(max<search[i]) max=search[i];
    }
    int cunchu[max+1],m=0;
    for(int i=0;i<=max;i++){
        string s1=to_string(i);
        string s2=to_string((int)pow(i,2));
        string s3=s2.substr(s2.size()-s1.size());
        if(s3==s1) m++;
        cunchu[i]=m;
    }
    for(int i=0;i<n;i++){
        if(i!=0) cout<<" ";
        cout<<cunchu[search[i]];
    }
    return 0;
}

第一次写博客，希望能给各位一个参考！