1040. 有几个PAT（25）

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？

输入格式：

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

输出格式：

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

输入样例：

APPAPT

输出样例：

2

#include<stdio.h>
int main(){
    char str[100005];
    int P[100005],T[100005],i,len,ans=0;
    scanf("%s",str);
    P[0]=str[0]=='P'?1:0;
    T[0]=str[0]=='T'?1:0;
    for(i=1;str[i];++i){
        P[i]=str[i]=='P'?P[i-1]+1:P[i-1];
        T[i]=str[i]=='T'?T[i-1]+1:T[i-1];
    }
    len = i;
    for(i=1;str[i+1];++i){
        if(str[i]=='A'){
            ans += P[i-1]*(T[len-1]-T[i]);
            ans %= 1000000007;
        }
    }
    printf("%d\n",ans);
    return 0;
}