C++/Python PAT乙级1091 N-自守数 (15分)

如果某个数 K 的平方乘以 N 以后，结果的末尾几位数等于 K，那么就称这个数为“N-自守数”。例如 3×92^2=25392，而 25392 的末尾两位正好是 92，所以 92 是一个 3-自守数。
本题就请你编写程序判断一个给定的数字是否关于某个 N 是 N-自守数。

输入格式：
输入在第一行中给出正整数 M（≤20），随后一行给出 M 个待检测的、不超过 1000 的正整数。

输出格式：
对每个需要检测的数字，如果它是 N-自守数就在一行中输出最小的 N 和 NK^2的值，以一个空格隔开；否则输出 No。注意题目保证 N<10。

输入样例：
3
92 5 233
  
输出样例：
3 25392
1 25
No

C++

#include<iostream>
#include<string.h>
#include<math.h>
#include<stdfix.h>
using namespace std;
int main()
{
    int m;
    cin >> m;
    for(int i=0;i<m;i++)
    {
        int k;
        cin >> k;
        string k1=to_string(k);
        int b1=k1.length();
        int a=k*k;
        int flag=0;
        for(int j=1;j<10;j++)
        {
            int a1=j*a,flag1=0;
            int aa=a1-k;
            string a2=to_string(aa);
            int b2=a2.length();
            for(int p=b2-1;p>=b2-b1;p--)
            {
                if(a2[p]!='0') {flag1=1;break;}
            }
            if(flag1==0)
            {
                cout<< j <<" "<< a1;
                flag=1;
                break;
            }
        }
        if(flag==0) cout <<"No";
        if(i<m-1) cout << endl;
    }
    return 0;
}

Python

m=int(input())
number=list(map(int,input().split()))

for i in range(m):
    k=number[i]
    flag=False
    for n in range(1,10):
        nk2=n*(k**2)
        if str(nk2)[-len(str(k)):]==str(k):
            flag=True
            print(n,nk2)
            break
    if not flag:
        print("No")