中南大学2018年ACM暑期集训前期训练题集(入门题) X: 又一道简单题

#include<iostream>
using namespace std;
int num[100000];
int main()
{
    int T;
    for(int i=32;i<100;i++){
        num[i*i]=1;
    }
    cin>>T;
    int n;
    int k=0;
    while(T--){
        cin>>n;
        k++;
        int a,b,c,d;
        a=n/1000;
        b=n/100%10;
        c=n/10%10;
        d=n%10;
        int ans=0;
        for(int i=0;i<=9;i++){
            if(a!=i&&i!=0&&num[i*1000+b*100+c*10+d]){ans++;}
            if(b!=i&&num[a*1000+i*100+c*10+d]){ans++;}
            if(c!=i&&num[a*1000+b*100+i*10+d]){ans++;}
            if(d!=i&&num[a*1000+b*100+c*10+i]){ans++;}
        }
        cout<<"Case "<<k<<": ";
        cout<<ans<<endl;
    }

}