[PAT乙级]1040 有几个PAT (思路+精简代码)

1040 有几个PAT

字符串 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

思路：如果暴力查找，对每个P，向右找到每个A和T，效率不优，且极可能超时。
考虑从数学的角度出发，对一个确定的P，能形成的PAT的个数为其右侧A的个数*T的个数。但鉴于我们从左端开始遍历，无法事先统计所有的A和T个数，这里不妨换一个计算对象，即对于一个确定的T，能形成的PAT的个数为其左侧P的个数*A的个数。现在你知道要怎么办了。
要想再提高运算效率，可以以PA的个数为统计的依据，也方便取模

#include <stdio.h>
int main() {
	int P = 0, PA = 0, PAT = 0;
	char ch;
	while ((ch = getchar()) != '\n') {
        if (ch == 'P') P++;
        else if (ch == 'A') PA = (PA + P) % 1000000007;	//发现A，增加P个PA，同时要取模
        else PAT = (PAT + PA) % 1000000007;	//发现T，PAT个数增加
    }
	printf("%d", PAT);
	return 0;
}