PAT乙级1040(C++)——龙哥哥的刷题路

1040 有几个PAT (25分)
字符串 APPAPT 中包含了两个单词 PAT，其中第一个 PAT 是第 2 位§，第 4 位(A)，第 6 位(T)；第二个 PAT 是第 3 位§，第 4 位(A)，第 6 位(T)。
现给定字符串，问一共可以形成多少个 PAT？

输入格式：
输入只有一行，包含一个字符串，长度不超过10^5
​​ ，只包含 P、A、T 三种字母。

输出格式：
在一行中输出给定字符串中包含多少个 PAT。由于结果可能比较大，只输出对 1000000007 取余数的结果。

输入样例：
APPAPT

输出样例：
2

解题思路：先统计每位对应的P的数量，然后再从最后一个位置开始遍历，统计出现的T的个数，每遇到一个A，便统计P的个数与T的个数的乘积，最后取模即可

#include <iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN = 100005;
char str[MAXN];
int p[MAXN] = { 0 };
int main() {
    scanf("%s",str);
	int length = strlen(str);
	int countT = 0;
	int total = 0;
	for (int i = 0; i < length; i++){
		if (i>0){
			p[i] = p[i - 1];
		}
		if (str[i] == 'P'){
			p[i]++;
		}
	}
	for (int i = length-1; i >=0; i--){
		if (str[i] == 'T'){
			countT++;
		}
		if (str[i] == 'A'){
			total = (total+p[i]*countT) % 1000000007;
		}
	}
	printf("%d\n", total);
	return 0;
}