PAT 1091 N-自守数

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

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

输入样例：

2

输出样例：

3 25392
1 25
No

代码如下（c++）：

#include <iostream>
#include <cmath>
using namespace std;
int judge(int,int);
int fun(int);
int main()
{
   int m,n;
    cin>>m;
    while(m--){
       cin>>n;
       int t = fun(n);//得到所需截取的位数
       int i,x;
       for(i=1;i<10;i++){
          x = i*n*n;
          int y = judge(x,t);
          int z = judge(y,t);//两个judge得到截取t位的数
          if(n==z) break;
       }
       if(i==10) cout<<"No"<<endl;
       else{
          cout<<i<<" "<<x<<endl;
       }
    }
    return 0;
}
int fun(int t){
    int i = 0;
    while(t){
       t=t/10;
       i++;
    }
    return i;
}
int judge(int x,int t){
   int sum = 0;
   while(t--){
     int y = x%10;
     x = x/10;
     sum = sum*10+y;
   }
   return sum;
}