PAT1003——走向ACM！

自从励志要参加ACM后，我就着手开始了准备。本来想通过做ZOJ来提升，无奈底子太薄，到了第二题FIRE NET就完全不行了。。。
但是发现了PAT后，发现这个中文OJ确实不错，所以准备拿它练手。希望各位大大对我的代码给出建议。谢谢大大们了。

---

1003. 我要通过！(20)

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于PAT的“答案正确”大派送 —— 只要读入的字符串满足下列条件，系统就输出“答案正确”，否则输出“答案错误”。

得到“答案正确”的条件是：

1. 字符串中必须仅有P, A, T这三种字符，不可以包含其它字符；
2. 任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
3. 如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a, b, c 均或者是空字符串，或者是仅由字母 A 组成的字符串。

现在就请你为PAT写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。
输入格式： 每个测试输入包含1个测试用例。第1行给出一个自然数n (<10)，是需要检测的字符串个数。接下来每个字符串占一行，字符串长度不超过100，且不包含空格。

输出格式：每个字符串的检测结果占一行，如果该字符串可以获得“答案正确”，则输出YES，否则输出NO。

---

代码如下：

#include
#include
#include

int main(){
    int cnt, yes = 0;   //用yes判断A所处的位置 
    int numA1 = 0, numA2 = 0, numA3 = 0;    //不同位置A的计数 
    scanf("%d", &cnt);
    getchar();
    char str[100];
    memset(str, 0, sizeof(str));

    for(int i = 1; i <= cnt; i++){
        gets(str);

        for(int k = 0; k < 100; k++){   //如果出现非PAT字符便直接打印NO 
            if(str[k] != 'P' && str[k] != 'A' && str[k] != 'T' && str[k] != 0){
                goto inputError;
            }
        }   

        for(int j = 0; j < 100; j++){   //对不同位置的A做出统计 
            if(str[j] == 'P'){
                yes = 1;
                continue;
            }
            if(str[j] == 'T'){
                yes = 2;
                continue;
            }
            if(str[j] == 0) break;

            if(yes == 0 && str[j] == 'A') numA1++;
            if(yes == 1 && str[j] == 'A') numA2++;
            if(yes == 2 && str[j] == 'A') numA3++;
        }       

        if(yes == 2 && (numA1 * numA2 == numA3) &&
             (numA1 != 0 || numA2 != 0 || numA3 != 0)) printf("YES\n"); 
                //条件很多：1.必须找到了T；2.P前PT中间的A之积等于T之后；3.A至少出现一次 
        else{
            inputError:
            printf("NO\n");
        }
        yes = 0;
        numA1 = 0; numA2 = 0; numA3 = 0;
        memset(str, 0, sizeof(str));
    }

    return 0;
}