PAT (Advanced Level) 1060. Are They Equal (25)

模拟题。坑点较多。

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<map>
#include<queue>
#include<string>
#include<stack>
#include<vector>
using namespace std;

int n;
string s,t;
struct X
{
    string f;
    int idx;
} ans[];

string change(string a)
{
    string res;
    if(a.length()-<n)
    {
        for(int i=a.length()-;i<n;i++) a=a+"";
        res=a;
    }

    else if(a.length()->n)
    {
        res="0.";
        for(int i=;i<n+;i++) res=res+a[i];
    }
    else res=a;
    return res;
}

string qu(string a)
{
    string res;
    if(a.length()==&&a[]=='') return a;

    int pos=-;
    for(int i=; i<a.length(); i++) if(a[i]=='.') pos=i;

    if(pos==-)
    {
        for(int i=; i<a.length(); i++)
        {
            if(a[i]=='') continue;
            else
            {
                for(int j=i; j<a.length(); j++) res=res+a[j];
                break;
            }
        }
    }
    else
    {
        for(int i=; i<pos-; i++)
        {
            if(a[i]=='') continue;
            else
            {
                for(int j=i; j<pos-; j++) res=res+a[j];
                break;
            }
        }
        for(int i=pos-; i<a.length(); i++) res=res+a[i];
    }

    if(res.length()==) res="";
    return res;
}

void R(int a)
{
    bool ling=;
    for(int i=; i<ans[a].f.length(); i++)
    {
        if(ans[a].f[i]=='.') continue;
        if(ans[a].f[i]!='') ling=;
    }

    if(ling) ans[a].idx=;
}

void PUT(int a)
{
    cout<<ans[a].f<<"*10^"<<ans[a].idx;
}

int main()
{
    scanf("%d",&n);
    cin>>s>>t;

    s=qu(s);
    t=qu(t);

    if(s[]==''&&s[]=='.')
    {
        int num=;
        for(int i=; i<s.length(); i++)
        {
            if(s[i]=='') num++;
            else
            {
                ans[].idx=-num;
                ans[].f="0.";
                for(int j=i; j<s.length(); j++) ans[].f=ans[].f+s[j];
                break;
            }
        }
        if(ans[].f.length()==)
        {
            ans[].f="0.0";
            ans[].idx=;
        }
    }

    else
    {
        int pos=-;
        for(int i=; i<s.length(); i++) if(s[i]=='.') pos=i;
        if(pos!=-)
        {
            ans[].f="0.";
            for(int i=; i<s.length(); i++) if(s[i]!='.') ans[].f=ans[].f+s[i];
            ans[].idx=pos;
        }
        else
        {
            ans[].f="0."+s;
            ans[].idx=s.length();
        }
    }

    ans[].f=change(ans[].f);

    if(t[]==''&&t[]=='.')
    {
        int num=;
        for(int i=; i<t.length(); i++)
        {
            if(t[i]=='') num++;
            else
            {
                ans[].idx=-num;
                ans[].f="0.";
                for(int j=i; j<t.length(); j++) ans[].f=ans[].f+t[j];
                break;
            }
        }
        if(ans[].f.length()==)
        {
            ans[].f="0.0";
            ans[].idx=;
        }
    }
    else
    {
        int pos=-;
        for(int i=; i<t.length(); i++) if(t[i]=='.') pos=i;
        if(pos!=-)
        {
            ans[].f="0.";
            for(int i=; i<t.length(); i++) if(t[i]!='.') ans[].f=ans[].f+t[i];
            ans[].idx=pos;
        }
        else
        {
            ans[].f="0."+t;
            ans[].idx=t.length();
        }
    }

    ans[].f=change(ans[].f);

    R(); R();

    if(ans[].f==ans[].f&&ans[].idx==ans[].idx)
    {
        printf("YES ");
        PUT();
    }

    else
    {
        printf("NO ");
        PUT();
        cout<<" ";
        PUT();
        cout<<endl;
    }

    return ;
}