PAT乙级 1003 我要通过！

1003 我要通过！

题目描述

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

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

字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c 均或者是空字符串，或者是仅由字母 A 组成的字符串。
现在就请你为 PAT 写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。

输入格式：

每个测试输入包含 1 个测试用例。第 1 行给出一个正整数 n (≤10)，是需要检测的字符串个数。接下来每个字符串占一行，字符串长度不超过 100，且不包含空格。

输出格式：

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

输入样例：

10
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA
APT
APATTAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO
NO
NO

方法

解题思路

题意理解：
首先要满足含有字Ｐ、Ａ、Ｔ，并且Ｐ、Ｔ只出现一次，并且Ｐ必须在Ｔ的前面；
然后类似于ｘPＡＴｘ这样的满足条件；
其次就是如果 aPbTc 是正确的，那么 aPbATca 也是正确的。可以得出推导如下结论：

举例：当PT之间只有一个A的时候： APATA —>> APAATAA , AAPATAA ----->> AAPAATAAAA ,…
当PT之间有两个A的时候：APAATAA ---->> APAAATAAA, AAPAATAAAA ---->> AAPAAATAAAAAA,…

分析：字符串分为三段，P的左边，P、T之间，T的右边，分别统计 A 的个数，计为cnt1,cnt2,cnt3；满足条件 cnt1*cnt2 = cnt3。

定义左右指针 left 和 right，
左指针找到起始字符 P，同时统计左边 A 的个数；
右指针找到结束字符 T，同时统计右边 A 的个数；
遍历左指针到右指针范围内的字符，同时统计中间 A 的个数；
最后判断是否 cnt1 * cnt2 == cnt3 。

代码

#include <iostream>
#include <cstdio>
using namespace std;
bool judge(string s)
{
    int cnt1 = 0, cnt2 = 0, cnt3 = 0;
    int left = 0, right = s.size() - 1;
    // 左指针扫描到字符为 P 停下来
    while(s[left] != 'P')
    {	
    	// 统计 P 左边 A 的个数
        if(s[left] == 'A')
            cnt1++;
        // 遇到其他字符返回 0
        else 
            return 0;
        left++;
    }
    // 右指针扫描到字符为 T 停下来
    while(s[right] != 'T')
    {
    	// 统计 T 右边 A 的个数
        if(s[right] == 'A')
            cnt3++;
        // 遇到其他字符返回 0
        else
            return 0;
        right--;
    }
   	// 统计 (left, right) 范围内 A 的个数
    for(int i = left + 1; i < right; i++)
    	if(s[i] == 'A')
        	cnt2++;
    // 如果这个范围内没有 A, 则不符合条件
    if(!cnt2)   return 0;
    else
    {
    	// 左边 A 的个数 * 中间 A 的个数 = 右边 A 的个数
    	if(cnt2 * cnt1 == cnt3)
        	return 1;
        else
            return 0;
    }
}
int main()
{
    int n;
    string s;
    cin >> n;
    while(n--)
    {
        cin >> s;
        if(judge(s))    cout << "YES" << endl;
        else    cout << "NO" << endl;
    }
    return 0;
}