PTA 1040 有几个PAT(c/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
结尾无空行

基本思路：找到A前面的P的个数和A后面的T的个数，将二者相乘，然后将所有满足上述条件的A所对应的乘积相加，即为最后的结果。

#include<stdio.h>
#include<iostream>
using namespace std;
int main()
{
	string str;
	cin >> str;
	int Pcount[100001] = { 0 };//统计从前到后每一个位置对应的P的个数
	int Tcount = 0;
	int sum = 0;
	for (int i = 0; i < str.length(); i++)
	{
		if (i > 0)
		{
			Pcount[i] = Pcount[i - 1];
		}
		if (str[i] == 'P')
		{
			Pcount[i]++;
		}
	}
	for (int i = str.length() - 1; i >= 0; i--)
	{
		if (str[i] == 'T')
		{
			Tcount++;
		}
		else if (str[i] == 'A')
		{
			sum= (sum + Pcount[i] * Tcount) % 1000000007;//Pcount[i]表示A前面的P的个数
		}
	}
	printf("%d", sum);
}

参考链接：
PTA 1040 有几个PAT