有几个PAT

1040. 有几个PAT（25）

时间限制
120 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CAO, Peng

字符串APPAPT中包含了两个单词“PAT”，其中第一个PAT是第2位(P),第4位(A),第6位(T)；第二个PAT是第3位(P),第4位(A),第6位(T)。

现给定字符串，问一共可以形成多少个PAT？

输入格式：

输入只有一行，包含一个字符串，长度不超过105，只包含P、A、T三种字母。

输出格式：

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

APPAPT

输出样例：

2

刚开始上来就一手暴力法，然后吃到了刷了50多个乙级题的第一发“TLE”。仔细想想发现顺序是固定的，A永远在中间，P在前，T在后。如果我知道一个A前面有多少个P后面多少个T做个乘法就可以，然后答案累加。那么首先统计T的数量，然后从头遍历，遇到P则P的数量+1,遇到T则T的总数-1，遇到A就乘法累加即可。代码如下：

#include<iostream>
#include<cstring>
using namespace std;
int main()
{
  string s1;
  int sum=0,sum1=0,sum2=0;
  bool flag=true;
  cin>>s1;
  for(int i=0;i<s1.length();i++)
    if(s1[i]=='T')sum++;
  for(int i=0;i<s1.length();i++)
  {
    if(s1[i]=='P')sum1++;
    if(s1[i]=='T')sum--;
    if(s1[i]=='A'){sum2+=(sum1*sum)%1000000007;sum2%=1000000007;}
  }
  cout<<sum2%1000000007<<endl;
}