PTA-1040 有几个PAT(题解)

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

这题是我好久以前做的，那时候刚开始的思路是直接暴力三重循环，但是根据样例来看，这样看显然不行，绝对会超时，于是乎我作罢。
如今偶然再次拾起，通过柳神也是知道了一种牛逼巧妙的解法思路，特此来写一篇题解。

<1>首先，我们既然要求PAT的个数，我们可以先想一想，要构成一个PAT，肯定三个都要顺序出现，令P，A，T字符的个数用变量P，A，T表示，刚开始个数都为0。
<2>然后，我们就可以这样想，先把所有T的个数先求出来，在从字符串第一个字符从前往后走，遇到一个P就加一，遇到一个T就减一，遇到一个A，就让总的PTA的个数加上这时候P与T个数的积就行，这样到最后就得到了最后的结果
<3>至于为什么要怎么做，我们可以这么想，假设有这么一个字符串PPATAT,
按照刚才的思路，我们可以先得出所有的T个数为2，那么接下来从第一个字符往后走就有，到P，P的个数加一，到A，就令总的个数sum加上目前P与T个数的积，这时候sum为4，也就是2乘2，这很巧妙，因为A的前后PT个数相乘就刚好为这个A能组成多少的个PAT。
<4>最后对1000000007取模就可以了

这是我的题解：

#include<bits/stdc++.h>
#include<algorithm>
using namespace std;

int main(){
	string str;cin>>str;int p=0,a=0,t=0,sum=0;
	for(int i = 0;i < str.size(); ++i){
		if(str[i]=='T'){
			t++;
		}
	}
	for(int i = 0;i < str.size();++i){
		if(str[i]=='P') p++;
		if(str[i]=='T') t--;
		if(str[i]=='A'){
			sum+=((p*t)%1000000007);
			sum%=1000000007;
		}
	} 
	cout<<sum;
}