1040 有几个PAT (25分)

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左边的P的个数和右边的T的个数，把问题转化为P和T的排列组合问题，即对于每一个A来说，一共有nP (p的个数) * nT (t的个数) 种可能，将每一个A的可能情况相加，即是总数。

代码如下： 

#include<bits/stdc++.h>
using namespace std;
int main()
{
	long long sumA[100005], cntP = 0, cntT = 0, cntA = 0, sum = 0;
	string a;
	getline(cin, a);
	for (int i = 0; i < a.length(); i++)//记录每一个A前P的个数 
	{
		if (a[i] == 'P')  cntP++;
		else if (a[i] == 'A')  sumA[++cntA] = cntP;
	}
	for (int i = a.length() - 1; i >= 0; i--)//记录每一个A后T的个数，并计算A前P的个数和A后T的个数的乘积 
	{
		if (a[i] == 'T')  cntT++;
		else if (a[i] == 'A'){
			sumA[cntA--] *= cntT;
			sum += sumA[cntA + 1];
		}		
	}printf("%d", sum % 1000000007);
}