PAT Basic Level 1003. 我要通过！(20)

【来源】

1003. 我要通过！(20)

【分析】

为了通过得到可爱的红色的答案正确需要满足一定的条件，具体判断时思路如下：

• 若有除了"PAT"以外的字符出现，答案错误；
• 若字符串中有"PAT"字符子串，则看"PAT"前后是否相等且均为空串或仅由'A'构成的字符串，如果是，答案正确，否则，答案错误；
• 若字符串中有'P'和'T'且'P'在'T'前面，那么整个字符串被分为3部分：头、中间和尾部。三个部分必须均为'A'构成的字符串或是空串。如果头部的长度等于中间的长度与尾部的长度之积，则答案正确，否则答案错误。

【源码】

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

bool onlyA(const string& s)
{
    for (int i = 0; i < s.size(); ++i)
    {
        if (s[i] != 'A'){
            return false;
        }
    }

    return true;
}

bool pass(const string& s)
{
    if (!(s.find('P') != string::npos && s.find('A') != string::npos 
        && s.find('T') != string::npos))
    {
        return false;
    }

    if (s.find("PAT") != string::npos)
    {
        size_t pos = s.find("PAT");
        string head = s.substr(0, pos);
        string tail = s.substr(pos + 3);
        if (head == tail && onlyA(head)){
            return true;
        }
        else{
            return false;
        }
    }

    size_t pPos = s.find('P');
    size_t tPos = s.find('T');

    if (pPos != string::npos && tPos != string::npos
        && pPos < tPos){
        string head = s.substr(0, pPos);
        string middle = s.substr(pPos+1, tPos-pPos-1);
        string tail = s.substr(tPos+1);

        if (onlyA(head) && onlyA(middle) && onlyA(tail)){
            if (tail.length() == middle.length() * head.length()){
                return true;
            }
        }
    }

    return false;
}

int main()
{
    int n;
    cin >> n;
    
    for (int i = 0; i < n; ++i)
    {
        string s;
        cin >> s;

        if (pass(s))
        {
            cout << "YES\n";
        }
        else
        {
            cout << "NO\n";
        }
    }
    return 0;
}

【点评】

此题考察稍复杂的字符串处理。第三个判断条件需要洞察出字符串的规律所在。