PAT B1040 有几个PAT

觉得题目还不错哦,看答案第一个想到KMP，可能是我练习太少，孤陋寡闻了`;.’

PAT B1040 有几个PAT （25 分）

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

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

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

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

• 输入样例：
  APPAPT

• 输出样例：
  2

#include< cstdio>
#include< cstring>
using namespace std;
const int maxn=100010;
const int MOD=1000000007;
int numLeftP[]={0};
int main(){
    char str[maxn];
    gets(str);
    int len=strlen(str);
    for(int i=0;i<len;i++){
        if(i!=0) numLeftP[i]=numLeftP[i-1];//注意此处处理0号元素的手法
        if(str[i]=='P') numLeftP[i]++;//妙的
    }
    int ans=0,rightNumT=0;
    for(int i=len-1;i>=0;i--){//从右往左累积T的个数，遇到A则结果增加numLeftP[i]*rightNumT
        if(str[i]=='T') rightNumT++;
        else if(str[i]=='A') ans=(ans+numLeftP[i]*rightNumT)%MOD;
    }
    printf("%d",ans);
    return 0;
}