PAT乙级 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。

二.题解

条件1：字符串只含有P、T、A三种字符。

条件2：xPATx 形式字符串均正确，说明初始阶段 PAT左边的A个数=PAT右边A的个数，并且字符串中有且只有一个P和T，字符串至少有一个A。

条件3：若aPbTc正确，则 aPbATca 也是正确的。观察前后字符串的变化，P和T中间每增加 一个A，字符串的末尾都会增加一个a字符串，不难发现，字符串中存在如下关系： 

              P左边A的个数 ✖ P、T中间A的个数=T右边A的个数

           

三.代码

参照csdn上OAOII的代码，增加注释并作部分修改后如下：

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int n;
    cin >> n;
    while (n--)
    {
        int a = 0, p = 0, t = 0;    //统计A，P，T的个数
        int loc_p, loc_t;           //loc_p显示P的位置，也可表示P左边字符的个数；loc_t可表示t的位置，也可表示t左边字符的个数
        char s[105];
        bool flag = true;
        scanf("%s", s);
        for (int i = 0; i < strlen(s); i++)
        {
            if (s[i] == 'P')
            {
                p++;
                loc_p = i;
            }
            else if (s[i] == 'A')
            {
                a++;
            }
            else if (s[i] == 'T')
            {
                t++;
                loc_t = i;
            }
            else
            {
                flag = false;
                break;
            }
        }
        //loc_p * (loc_t - loc_p - 1) == strlen(s) - loc_t - 1表示p左边A的个数✖p、t中间A的 
        个数等于t右边A的个数
        if (flag == true && p == 1 && t == 1 && a > 0 && (loc_t-loc_p > 1) && (loc_p * (loc_t - loc_p - 1) == strlen(s) - loc_t - 1))
        {
            cout << "YES" << endl;
        }
        else
        {
            cout << "NO" << endl;
        }
    }
}