【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个数 = 左侧 $P$ 字符个数 * 右侧$T$字符个数。

1. 变量 $p$ 为左侧 $P$ 字符个数，变量 $t$ 为左侧 $T$ 字符个数，变量 $lent$ 为 $T$ 字符总数，$lent-t$ 为右侧$T$字符个数，$sum$ 为PAT 总数
2. 遍历字符串，保存 $T$ 字符总数为 $lent$，再次遍历字符串，遇到 $P$ 执行 $p++$，遇到 $T$ 执行 $t++$，遇到 $A$ 计算该位置 $A$ 的PAT个数为 $p\ast (lent-t)$

$Note$: 字符长度不超过 $10{}^5$，所以采用 $long$ $long$ 类型存储

答案代码：

#include <iostream>
#include <string>
using namespace std;

#define MAXN 1000000007
int main() {
	string s;
	cin >> s;
	
	long long lent = 0;
	for (long long i = 0; i < s.size(); i++)
		if (s[i] == 'T') lent++;
		
	long long p = 0, t = 0, sum = 0;
	for (long long i = 0; i< s.size(); i++) {
		if (s[i] == 'P') p++;
		if (s[i] == 'T') t++;
		if (s[i] == 'A') {
			sum = (sum + (p*(lent-t)) % MAXN) % MAXN;
		}
	} 
	cout << sum << endl;

	return 0;
}