PAT乙级题解——1040 有几个PAT (25分)

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

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

输入格式：
输入只有一行，包含一个字符串，长度不超过10
​5
​​ ，只包含 P、A、T 三种字母。

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

输入样例：

APPAPT

输出样例：

2

每个确定的A字符所能组成的PAT数量 = 该A左侧的P数量 X 该A右侧的T数量；总的PAT数量等于所有A所能组成的PAT数量之和

#include <iostream>
#include<map>
const int MOD = 1000000007;
using namespace std;

int main(){
    string str;
    cin >> str;
    int pnum = 0,tnum = 0,pat = 0;
    for (int i = 0; i < str.length(); i++) {
        if (str[i] == 'T') {
            tnum++;
        }
    }
    for (int j = 0; j < str.length(); j++) {
        if (str[j] == 'P') {
            pnum++;
        }
        if (str[j] == 'A') {
            pat = (pat + (pnum * tnum) % MOD) % MOD;
        }
        if (str[j] == 'T') {
            tnum--;
        }
    }
    cout << pat << endl;
    return 0;
}