PAT/PATBasic/1003.c

/**
 * 1003. 我要通过！
 * “答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于PAT的“答案正确”大派送
 *  —— 只要读入的字符串满足下列条件，系统就输出“答案正确”，否则输出“答案错误”。
 * 
 * 得到“答案正确”的条件是：
 * 
 * 1. 字符串中必须仅有P, A, T这三种字符，不可以包含其它字符；
 * 2. 任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者
 * 是仅由字母 A 组成的字符串；
 * 3. 如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a, b, c 均或者是空字符
 * 串，或者是仅由字母 A 组成的字符串。
 * 
 * 现在就请你为PAT写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。
 * 输入格式： 每个测试输入包含1个测试用例。第1行给出一个自然数n (<10)，是需要检测
 * 的字符串个数。接下来每个字符串占一行，字符串长度不超过100，且不包含空格。
 * 
 * 输出格式：每个字符串的检测结果占一行，如果该字符串可以获得“答案正确”，则输出
 * YES，否则输出NO。
 * 
 * 输入样例：
 * 8
 * PAT
 * PAAT
 * AAPATAA
 * AAPAATAAAA
 * xPATx
 * PT
 * Whatever
 * APAAATAA
 * 输出样例：
 * YES
 * YES
 * YES
 * YES
 * NO
 * NO
 * NO
 * NO
 * 
 * (暂未全部通过)
 **/
#include <stdio.h>
#define YES 1
#define NO  0

int main()
{
    char c;
    int num;
    int YESorNO;
    scanf("%d", &num);
    getchar();
    
    for(int i = 0; i < num; i++)
    {
        YESorNO = YES;
        /* three numbers to count number of As: before P, between P&T and after T */
        int count[3] = {0, 0, 0};
        /* index of the array above */
        int pos = 0;
        while((c = getchar()) != '\n')
        {
            if(c == 'A')
            {
                count[pos]++;
            }
            else if(c == 'P' && pos == 0)
            {
                pos = 1;
            }
            else if(c == 'T' && pos == 1)
            {
                pos = 2;
            }
            else
            {
                YESorNO = NO;
            }
        }
        /*
         * If the string is "correct", the number of As from
         *   before P, 
         *   between P&T and 
         *   after T
         * should satisfy the following relationship:
         *   (between - 1) * before <= after
         * 
         * Because according to the rule No.3, if a string is "correct", we insert
         * an A before T, and copy everything before P at the end of the string,
         * the new string will still be "correct", thus getting the relation above.
         **/
        if(YESorNO == YES && pos == 2 && 
            (count[1] == 1 || 
            (count[1] > 1 && count[2] >= count[0] * (count[1] - 1))))
        {
            puts("YES");
        }
        else
        {
            puts("NO");
        }
    }
    return 0;
}