PAT (Advanced Level) 1100. Mars Numbers (20)

简单题。

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

char a[][]={
    "tret","jan", "feb", "mar", "apr", "may", "jun", "jly", "aug", "sep", "oct", "nov", "dec"
};

char b[][]={
    "zzz","tam", "hel", "maa", "huh", "tou", "kes", "hei", "elo", "syy", "lok", "mer", "jou"
};

int main()
{
    int n; char s[];
    scanf("%d",&n); getchar();
    for(int i=;i<=n;i++)
    {
        gets(s);
        if(s[]>=''&&s[]<='')
        {
            int num=;
            for(int i=;s[i];i++) num=num*+s[i]-'';
            if(num<=) printf("%s\n",a[num]);
            else if(num%==) printf("%s\n",b[num/]);
            else printf("%s %s\n",b[num/],a[num%]);
        }
        else
        {
            int p=-;
            for(int i=;s[i];i++) if(s[i]==' ') p=i;
            if(p==-)
            {
                int num=;
                for(int i=;i<=;i++)
                    if(strcmp(a[i],s)==) num=i;
                for(int i=;i<=;i++)
                    if(strcmp(b[i],s)==) num=*i;
                printf("%d\n",num);
            }
            else
            {
                int num=;
                char op[]; int sz=;
                for(int i=;i<p;i++) op[sz++]=s[i]; op[sz]=;
                for(int i=;i<=;i++)
                    if(strcmp(b[i],op)==) num=*i;
                sz=;
                for(int i=p+;s[i];i++) op[sz++]=s[i]; op[sz]=;
                for(int i=;i<=;i++)
                    if(strcmp(a[i],op)==) num+=i;
                printf("%d\n",num);
            }
        }
    }
    return ;
}