PTA 1003.我要通过！代码和关键点

PTA 1003.我要通过！代码和关键点

题目

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

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

1.字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
2.任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
3.如果 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
结尾无空行

关键点

左侧A的数量*中间A的数量=右侧A的数量

这个条件是怎么得到的呢？我们要对题干的第二条、第三条进行梳理和归纳：

条件二规定形如xPATx的串合法，在此基础上套用条件三：

如果aPbTc是正确的，那么aPbATca也是正确的，其中a、b、c均或者是空字符串，或者是仅由字母A组成的字符串。
可知xPAATxx也合法(例如：PAAT[x=’’]、APAATAA[x=‘A’]、APAAATAAA[x=‘AA’])，继续归纳总结：xPAAATxxx合法(例如：APAAATAAA[x=‘A’]、AAAPAAATAAAAAAAAA[x=‘AAA’])…

合法字符串确实有规律可循：对于字符串"[A1]P[A2]T[A3]"([An]代表含n个由’A’组成的子串)，当A1*A2=A3，且满足条件①时，则判定合法。

且：[A2]中，A的个数必须大于零

AC c代码

在#include <iostream>
#include <string>
#include <cstring>
using namespace std;

int main()
{
    int n, i;
    int tpos, ppos;

    cin >> n;
    string s;
    while (n--)
    {
        cin >> s;
        int pnum = 0, tnum = 0, a = 0, b = 0, c = 0, other = 0;
        for (i = 0; i < s.length(); i++)
        {
            if (s[i] == 'T')
            {
                tnum++;
                tpos = i;
            }
            else if (s[i] == 'P')
            {
                pnum++;
                ppos = i;
            }
            else if (s[i] == 'A')
            {
                if (pnum == 0)
                    a++; //p前面a的数量
                else if (pnum == 1 && tnum == 0)
                    b++; //p，t中间a的数量
                else if (tnum == 1)
                    c++; //t后面的a的数量
            }
            else
                other++;
        }
        if (tnum == 1 && pnum == 1 && other == 0 && tpos - ppos > 0 && (a > 0 || b > 0 || c > 0))
        {
            if ((a * b == c) && b > 0)
                cout << "YES\n";
            else
                cout << "NO\n";
        }
        else
            cout << "NO\n";
    }
    return 0;
}

参考文章：

1. 关键点思路来源
2. AC cpp代码思路来源，不过它的代码有误，我修改了一下