PAT BASIC LEVEL 1040. 有几个PAT（25）

1040. 有几个PAT（25）

字符串APPAPT中包含了两个单词“PAT”，其中第一个PAT是第2位(P),第4位(A),第6位(T)；第二个PAT是第3位(P),第4位(A),第6位(T)。
现给定字符串，问一共可以形成多少个PAT？

输入格式：

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

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

输入样例：
APPAPT

输出样例：
2

Answer:

#include<iostream>
#include<vector>
using namespace std;
int main() {
    long long count = 0;
    char input[100001];
    cin >> input;
    int i = 0;
    long long p = 0, a = 0;
    while(input[i]) {
        while(input[i]=='P') {
            p++;
            i++;
        }
        while(input[i]=='A') {
            a+=p;
            i++;
        }
        while(input[i]=='T') {
            count += a;
            i++;
        }
    }
    cout << count%1000000007;
}

PS.
本来想要是能一下五个问题都一次通过就好了。
然而运行超时了。虽然运行超时也在意料之中吧。
所以改了算法。
好像又学到了什么。
笑。