1093 Count PAT‘s (25 分)

题目链接

题意

给出一个仅包含P A T的字符串，可以组成多少个PAT子串。

思路

PAT中A处于中间，所以对于字符串中的每个A，都可以和它前面的所有P和后面的所有T组成p*t个PAT子串，因此只需找出每个A前P的个数和P后T的个数即可。
从前往后遍历字符串，对于遇到的每个A，对应P的个数为已遍历P的个数，T的个数为T的总个数减已遍历T的个数，两者相乘即为这个A可组成的PAT子串个数。注意要对1000000007取模。

代码

#include <bits/stdc++.h>
using namespace std;
int main()
{
    string s;
    cin>>s;
    int p=0,a=0,t=0,ans=0;
    for(int i=0;i<s.size();i++){
        if(s[i]=='T') t++;
    }
    for(int i=0;i<s.size();i++){
        if(s[i]=='P') p++;
        if(s[i]=='T') t--;
        if(s[i]=='A') ans=(ans+(p*t)%1000000007)%1000000007;
    }
    printf("%d",ans);
}