PAT B1040 有几个PAT

问题描述

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

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

输入格式：

输入只有一行，包含一个字符串，长度不超过10^(​5)，且只包含 P、A、T 三种字母。

输出格式：

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

输入样例：

APPAPT

输出样例：

2

总结

如果一个A左边有x个P右边有y个T，那么这个A对应有x*y个PAT组合，遍历字符串，依次累加，得到最后结果

AC代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>


int main()
{
    //freopen("input.txt","r",stdin);//用于读入input.txt中的数据
    //freopen("output.txt","w",stdout);//用于将输出结果写到output.txt中,cmd窗口中不显示结果
    char s[100001];
    scanf("%s",s);
    int len=strlen(s);
    int numofleftP[len];
    int numofrightT=0;
    memset(numofleftP,0,len);
    long long totalnum=0;
    for(int i=1;i<len;i++)
    {//计算左边有多少个P
        if(s[i-1]=='P')
        {
            numofleftP[i]=numofleftP[i-1]+1;
        }
        else
        {
            numofleftP[i]=numofleftP[i-1];
        }
    }
    for(int i=len-2;i>=0;i--)
    {//
        if(s[i+1]=='T')numofrightT++;
        if(s[i]=='A')totalnum+=numofrightT*numofleftP[i];
    }
    printf("%d",totalnum%1000000007);//这里输出格式要填%d，pat才能通过，如果填写%I64d，会报格式错误

}