PAT (Basic Level) Practice 1040

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;
}