【PAT乙级】1091 N-自守数

问题描述
如果某个数 K 的平方乘以 N 以后，结果的末尾几位数等于 K，那么就称这个数为“N-自守数”。例如 $3\times 92^2=25392$，而 25392 的末尾两位正好是 92，所以 92 是一个 3-自守数。

本题就请你编写程序判断一个给定的数字是否关于某个 N 是 N-自守数。

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

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

样例输入
3
92 5 233

样例输出
3 25392
1 25
No

C++代码

#include<bits/stdc++.h>
using namespace std;

int judge(int k,int nk2){
    int a[10],i=0,j=0;
    while(nk2!=0){
        a[i++]=nk2%10;
        nk2/=10;
    }
    while(k!=0){
        if(k%10!=a[j++]) return 0;
        k/=10;
    }
    return 1;
}

int main(){
    int m;
    cin>>m;
    while(m--){
        int k,flag=0;
        cin>>k;
        for(int n=1;n<10;n++){
            long long int nk2=n*k*k;
            if(judge(k,nk2)==1){
                printf("%d %lld\n",n,nk2);
                flag=1;
                break;
            }
        }
        if(flag==0) printf("No\n");
    }
    return 0;
}