PAT乙级1003

1003. 我要通过！(20)

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于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<iostream>
#include<string>

using namespace std;

bool isRight(const string&);
bool isFirst(const string&);
bool isSecond(const string&);
bool isThird(const string&);
bool isAllAL(const string&,const int);
bool isAllAR(const string&, const int);

int main()
{
    string str1;
    int num = 0;
    cin >> num;
    for (int i = 0; i < num; i++)
    {
        cin >> str1;
        if (isRight(str1))
            cout << "YES\n";
        else
            cout << "NO\n";
    }

    system("pause");
    return 0;
}

bool isRight(const string& str)
{
    if (isFirst(str))
        if (isSecond(str) || isThird(str))
            return true;
    return false;
}

bool isFirst(const string& str)
{
    for (int i = 0; i < str.size(); i++)
        if (str[i] != 'P' && str[i] != 'A' && str[i] != 'T')
            return false;
    return true;
}

bool isSecond(const string& str)
{
    for (int i = 0; i < str.size() - 2; i++)
        if (str[i] == 'P' && str[i + 1] == 'A' && str[i + 2] == 'T')
            if (isAllAL(str, i) && isAllAR(str, i))
                return true;
    return false;
}

bool isThird(const string& str)
{
    for (int i = str.size() - 1; i >= 0; i--)
    {
        if (str[i] == 'P')
        {
            for (int j = i + 1; j < str.size() - 1; j++)
                if (str[j] == 'A' && str[j + 1] == 'T')
                    if (isAllAL(str, i) && isAllAR(str, j + 1))
                        return true;
            break;
        }
    }
    return false;
}

bool isAllAL(const string& str,const int x)
{
    for (int i = x - 1; i >= 0; i--)
        if (str[i] != 'A')
            return false;
    return true;
}

bool isAllAR(const string& str, const int x)
{
    for (int i = x+1; i <str.size(); i++)
        if (str[i] != 'A')
            return false;
    return true;
}

引用代码：

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

int func(string str);

int main() {
    int n;
    cin >> n;
    getchar(); //读入n后要用getchar读走空格
    string *s = new string [n];
    for (int i = 0; i < n; i++) {
        getline(cin, s[i]); //读入一整行的字符串，用getline(cin, string x);
    }
    int *t = new int [n];
    for (int i = 0; i < n; i++) {
        t[i] = 0;
    }
    for (int j = 0; j < n; j++) {
        t[j] = func(s[j]);
    }
    
    for (int k = 0; k < n; k++) {
        if (t[k] == 1) {
            cout << "YES" << endl;
        }
        if (t[k] == 0) {
            cout << "NO" << endl;
        }
    }
    delete [] s;
    delete [] t;
    return 0;
}

int func(string str) {
    int len = str.length();
    if (len <= 2)
        return 0;
    for (int i = 0; i < len; i++) {
        if (str[i] != 'P' && str[i] != 'A' && str[i] != 'T') {
            return 0;
        }
        if (str[i] == 'P') {
            // 找到P之后，P前面的所有字母只能是A，如果不是A，就return 0;
            for (int j = 0; j < i; j++) {
                if (str[j] != 'A')
                    return 0;
            }
            // P之后的一个字母也必然是A，如果不是，直接return 0;
            if (str [i + 1] != 'A')
                return 0;

            for (int k = i + 2; k < len; k++) {
                if (str[k] != 'A') {
                    // P后面若干个A，如果A结束后的第一个字母不是T，不符合直接return 0;
                    if (str[k] != 'T') {
                        return 0;
                    }
                    // k是当前T的位置 如果T后面的字符长度不是P前面字符长度的 k - 1 - i倍，错误直接return 0;
                    if ((len - 1 - k) != ((k - 1 - i) * i))
                        return 0;
                    // 如果T后面的字母有不是A的。直接return 0;
                    for (int m = k + 1; m < len; m++) {
                        if (str[m] != 'A')
                            return 0;
                    }
                }
            }
        }
    }
    // 如果有幸到了这一步，那就是正确的啦。return 1;
    return 1;

值得学习的地方：

1、利用堆区开辟确定数组的大小

2、采用string数组存下每一个需要检验的数（作为for中调用函数的参数）

3、用一个for循环为数组初始化