1040 有几个PAT(25)(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
分析:PAT的数量为每一个A之前P的数量 * A之后T的数量之和。所以题目转化为求每一个A之前P的数量和每一个A之后T的数
量。 首先将字符串中所有T的数量统计出来,之后再从头开始遍历,记录P的数量,并且遍历到T时减去T的数量,直到遇到A时计算当前A所对应的PAT的数量。
代码:
#include<iostream>
#include<string>
using namespace std;
int main() {
string s;
int P = 0, T = 0;
int ans = 0;
cin >> s;
for (int i = 0; i < s.size(); i++) {
if (s[i] == 'T') {
T++;
}
}
for (int i = 0; i < s.size(); i++) {
if (s[i] == 'P') {
P++;
}
if (s[i] == 'T') {
T--;
}
if (s[i] == 'A') {
ans = (ans + P * T) % 1000000007;
}
}
cout << ans << endl;
return 0;
}