PAT乙级 1003 我要通过！ (20 分) C语言

一、题目

二、源代码

#include<stdio.h>
int main()
{
    int n;
    scanf("%d", &n);
    while(n--)
    {
        char str[101];
        scanf("%s", str);
        int i=0,a=0,b=0,c=0,pcount=0,tcount=0;
        while(str[i]!='\0'&& (str[i]=='P' || str[i]=='A' || str[i]=='T'))
        {
            if(str[i]=='A')
            {
                if(pcount==0)a++;
                else
                {
                    if(tcount==0)b++;
                    else c++;
                }
            }
            else if(str[i]=='P')
            {
                pcount++;
                if(pcount>1)break;
            }
            else 
            {
                tcount++;
                if((a==0&&b==0)||pcount==0||tcount>1)break;
            }
            i++;
        }
        if(str[i]=='\0' && tcount==1 && pcount==1 && b>0 && a*b==c)printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}

三、运行结果