1040. 有几个PAT（25）

AI那点小事

发布于 2020-04-20 17:46:11

3590

发布于 2020-04-20 17:46:11

文章被收录于专栏：AI那点小事

字符串APPAPT中包含了两个单词“PAT”，其中第一个PAT是第2位(P),第4位(A),第6位(T)；第二个PAT是第3位(P),第4位(A),第6位(T)。

现给定字符串，问一共可以形成多少个PAT？

输入格式：

输入只有一行，包含一个字符串，长度不超过105，只包含P、A、T三种字母。

输出格式：

在一行中输出给定字符串中包含多少个PAT。由于结果可能比较大，只输出对1000000007取余数的结果。

输入样例： APPAPT 输出样例： 2

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

int main()
{
    string str;
    cin>>str;
    int P = 0;
    int PA = 0;
    int PAT = 0;
    for (int i = 0 ; i < str.length() ; i++){
        if ( str[i] == 'P'){
            P++;
        }
        if ( str[i] == 'A'){
            PA += P;
            PA %= 1000000007;
        }
        if ( str[i] == 'T'){
            PAT += PA;
            PAT %= 1000000007;
        } 
    }
    cout<<PAT;

    return 0;
}