1040 有几个PAT (25分)

 

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

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

输入格式:

输入只有一行,包含一个字符串,长度不超过10​5​​,只包含 PAT 三种字母。

输出格式:

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

输入样例:

APPAPT

输出样例:

2
#include "stdio.h"
#include "string.h"
#define MOD 1000000007
char str[100001];
int leftP[100001],rightT;
int main(){
    gets(str);
    int len = strlen(str);
    for (int i = 0; i < len; ++i) {
        if(str[i]=='P'){
            leftP[i]++;
        }
        leftP[i+1] = leftP[i];
    }
    int ans = 0;
    for (int i = len-1; i >= 0; i--) {
        if(str[i]=='T'){
            rightT++;
        }else if(str[i]=='A'){
            ans = (ans+leftP[i]*rightT)%MOD;
        }
    }
    printf("%d",ans);
}

 

你可能感兴趣的:(pat乙级)