CSU 1785 又一道简单题

题意：给定一个四位数，问是否能改变一位数之后（最高位不能为0）使之成为一个数的平方和。
思路：直接开方模拟

#include<cstdio>
#include<queue>
#include<iostream>
#include<vector>
#include<map>
#include<cstring>
#include<cmath>
#include<string>
#include<set>
#include<stack>
#include<algorithm>
#define cle(a) memset(a,0,sizeof(a))
#define inf(a) memset(a,0x3f,sizeof(a))
#define ll long long
#define Rep(i,a,n) for(int i=a;i<=n;i++)
using namespace std;
const int INF = ( 2e9 ) + 2;
int main()
{
    int t;
    scanf("%d",&t);
    int Cas=0;
    while(t--)
    {
        int n;
        char s[5],bak[5];
        scanf("%s",&s);
        for(int i=0;i<4;i++)bak[i]=s[i];
        n=0;
        for(int i=0;i<4;i++)
        n=n*10+s[i]-'0';
        int f=0;
        double c= sqrt(1.0*n);
        int d=(int)c;
        if(c==d)f=1;
        int ans=0;
        for(int i=1;i<=9;i++)
        {
            s[0]='0'+i;
            int t=0;
            for(int i=0;i<4;i++)
            t=t*10+s[i]-'0';
            double test=sqrt(1.0*t);
            int d=(int)test;
            if(d==test)ans++;
        }
        for(int i=1;i<=3;i++)
        {
            for(int i=0;i<4;i++)s[i]=bak[i];
            for(int j=0;j<=9;j++)
            {
                s[i]='0'+j;
                int t=0;
                for(int i=0;i<4;i++)
                t=t*10+s[i]-'0';
                double test=sqrt(1.0*t);
                int d=(int)test;
                if(d==test)ans++;
            }
        }
        if(f)ans-=4;
        printf("Case %d: %d\n",++Cas,ans);
    } 
}