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？

输入格式：

输入只有一行，包含一个字符串，长度不超过105，只包含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.

本来想要是能一下五个问题都一次通过就好了。

然而运行超时了。虽然运行超时也在意料之中吧。

所以改了算法。

好像又学到了什么。

笑。