PAT (Basic Level) Practice （中文）1091 N-自守数

#include<stdio.h>
int fun(int K)
{
    int wei=fun_1(K);
  //  printf("wei==%d\n",wei);
    int N;
    int l=1;
    for(N=1;N<10;N++)
    {
        int p=fun_2(N*K*K,wei);
       // printf("p==%d\n",p);
        if(p==K) {
            l=0;
            //printf("%d %d\n",N,N*K*K);
            break;
        }
    }
    if(l==1) return -1;
    else return N;
}
int fun_1(int n)
{
    int temp=0;
    while(n)
    {
        n/=10;
        temp++;
    }
    return temp;
}

int fun_2(int n,int wei)
{
    int i=1;
    int t=0;
    int sum=0;
    int *arr=(int*)malloc(sizeof(int)*wei);
    while(i<=wei)
    {
        int temp=n%10;
        arr[t++]=temp;
        n/=10;
        i++;
    }
    for(i=t-1;i>=0;i--)
        sum=sum*10+arr[i];
    return sum;
}
int main()
{
    int M;
    int Ki[22];
    int i;
    scanf("%d",&M);
    for(i=0;i<M;i++)
    {
        scanf("%d",&Ki[i]);
    }
    for(i=0;i<M;i++)
    {
        int t=fun(Ki[i]);
        if(t<0) printf("No\n");
        else printf("%d %d\n",t,t*Ki[i]*Ki[i]);
    }
    return 0;
}