又一道简单题

又一道简单题
Time Limit: 5000MS Memory Limit: 65535KB 64bit IO Format:
Submit

Status

Description
输入一个四个数字组成的整数 n,你的任务是数一数有多少种方法,恰好修改一个数字,把它 变成一个完全平方数(不能把首位修改成 0)。比如 n=7844,有两种方法:3844=62^2和 7744=88^2。

Input
输入第一行为整数 T (1<=T<=1000),即测试数据的组数,以后每行包含一个整数 n (1000<=n<=9999)。
Output
对于每组数据,输出恰好修改一个数字,把 n 变成完全平方数的方案数。
Sample Input
2
7844
9121
直接暴力遍历所有情况

#include <iostream>
#include <cstring>
using namespace std;
int vist[10010],used[10010];
int a[10];
void check()
{
    for(int i=32;i<100;i++)
    vist[i*i]=1;
}
int deal(int i,int j)
{
    int sum=0;
    for(int k=1;k<=4;k++)
    {
        if(k==i)
        sum=sum*10+j;
        else
        sum=sum*10+a[k];
    }
    return sum;
}
int main()
{
    int c=1,t;
    cin>>t;
    memset(vist,0,sizeof(vist));
    check();
    while(t--)
    {
        char str[10];
        cin>>str;
        memset(used,0,sizeof(used));
        for(int i=0;i<4;i++)
        a[i+1]=str[i]-'0';
        int ans=0;
        int val;
        for(int i=1;i<=4;i++)
        {
            for(int j=0;j<=9;j++)
            {
                if(i==1&&j==0) continue;
                if(a[i]==j) continue;
                val=deal(i,j);
                if(used[val])continue;
                if(vist[val])ans++;
                used[val]=1;
            }
        }
        cout<<"Case "<<c++<<": "<<ans<<endl;
    }
    return 0;
}