PAT (Basic Level):1040 有几个PAT（25）

题目信息

字符串APPAPT中包含了两个单词“PAT”，其中第一个PAT是第2位(P),第4位(A),第6位(T)；第二个PAT是第3位(P),第4位(A),第6位(T)。
现给定字符串，问一共可以形成多少个PAT？
输入格式：
输入只有一行，包含一个字符串，长度不超过10⁵，只包含P、A、T三种字母。
输出格式：
在一行中输出给定字符串中包含多少个PAT。由于结果可能比较大，只输出对1000000007取余数的结果。
输入样例：
APPAPT
输出样例：
2

第一次 运行超时的算法

#include
using namespace std;
int main(){
    string s;
    cin >> s;
    int count=0;
    for(int i=0;i

分析

须注意，此题的每次计算结果都要取余

最终代码

#include
using namespace std;
int main(){
    string s;
    cin >> s;
    int res=0,i=0,j,pnum=0,tnum=0;
    while(i0){
        if(s[i]=='P') pnum++;
        if(s[i]=='T') tnum--;
        if(s[i]=='A') res=(res+pnum*tnum)%1000000007;
        i++;
    }
    printf("%d",res);
    return 0;
}

测试结果

image.png