Gym 101653T Runes (模拟）

给你一个字符串 是[number][op][number]=[number]的形式， 其中op可以为+- *

数字其中有些‘？’问你能不能用0~9中的数字替代‘？’使等式成立

如果可能的话输出最小的 否则输出-1

字符串中已经出现的字符不能使用 不能有前导0和-0

乘法可能会爆int 所以要开long long

要注意几组极端的的样例

?*?=?   输出0  0*0=0 +-也是一样

?*?=?? 输出-1  不能有前导0

-?*?=? 输出-1  不过数据中没有-0的情况 不考虑也可以AC

感觉自己的代码能力还是不强  模拟题想的慢实现的慢而且写的不是很简单

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<queue>
#include<map>
#include<algorithm>
#include<iostream>
#define ll long long
#define ull unsigned long long
const int N=1e9+7;
const int M=1e3+5;
using namespace std;

int e,g,f[15],i,j;
int flag1,flag2,flag3,l1,l2,l3,ans;
ll q1,q2,q3;
char a[M],b[M],c[M],d[M],fh;

void init()   //初始化
{
    memset(f,0,sizeof(f));
    flag1=1,flag2=1,flag3=1,l1=0,l2=0,l3=0;
    ans=-1;
}

void solve_f()  //表达式中已经出现的数字不能使用
{
    e=strlen(a);
    for(i=0; i<e; i++)
    {
        if(a[i]>='0'&&a[i]<='9')
        {
            g=a[i]-'0';
            f[g]=1;
        }
    }
}

void solve_bcd()  //把输入的字符串拆成b c d和+-*
{
    for(i=0; i<e; i++)
    {
        if(flag2)
        {
            if((a[i]>='0'&&a[i]<='9')||(a[i]=='?')) flag1=0;
            if(!flag1)
            {
                if(a[i]=='-'||a[i]=='+'||a[i]=='*')
                {
                    fh=a[i];
                    flag2=0;
                    continue;
                }
            }
            b[l1++]=a[i];
        }
        else
        {
            if(flag3)
            {
                if(a[i]=='=')
                {
                    flag3=0;
                    continue;
                }
                c[l2++]=a[i];
            }
            else
            {
                d[l3++]=a[i];
            }
        }
    }
}

void solve_ans()
{
    for(i=0; i<=9; i++)
    {
        if(f[i]) continue;       //表达式中已经出现的数字则直接continue
        q1=0,q2=0,q3=0;
        if(i==0)  //如果出现前导0和-0 continue
        {
            if(b[0]=='?'&&l1>1) continue;
            if(c[0]=='?'&&l2>1) continue;
            if(d[0]=='?'&&l3>1) continue;
            if(b[0]=='-'&&b[1]=='?') continue;
            if(c[0]=='-'&&c[1]=='?') continue;
            if(d[0]=='-'&&d[1]=='?') continue;
        }
        for(j=0; j<l1; j++)  //求第一个数
        {
            if(b[j]=='-') continue;
            if(b[j]>='0'&&b[j]<='9')
            {
                g=b[j]-'0';
                q1=q1*10+g;
            }
            if(b[j]=='?') q1=q1*10+i;
        }
        if(b[0]=='-') q1=-q1;
        for(j=0; j<l2; j++) //求第二个数
        {
            if(c[j]=='-') continue;
            if(c[j]>='0'&&c[j]<='9')
            {
                g=c[j]-'0';
                q2=q2*10+g;
            }
            if(c[j]=='?') q2=q2*10+i;
        }
        if(c[0]=='-') q2=-q2;
        for(j=0; j<l3; j++) //求第三个数
        {
            if(d[j]=='-') continue;
            if(d[j]>='0'&&d[j]<='9')
            {
                g=d[j]-'0';
                q3=q3*10+g;
            }
            if(d[j]=='?') q3=q3*10+i;
        }
        if(d[0]=='-') q3=-q3;
        if(fh=='+'&&q1+q2==q3)      ans=i;
        if(fh=='-'&&q1-q2==q3)      ans=i;
        if(fh=='*'&&q1*q2==q3)      ans=i;
        if(ans!=-1) break;
    }
    printf("%d\n",ans);
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",a);
        init();
        solve_f();
        solve_bcd();
        solve_ans();
    }
    return 0;
}