PAT乙级1040 有几个PAT（C语言）

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

分析一

单独用一次循环，统计T的个数，在用循环查找A，同时记录A之前P的个数，减去A之前出现的T,PAT个的个数即为两数乘积

代码一

#include<stdio.h>
int main()
{
    char a[100005];
    scanf("%s",a);//读取字符串
    int i,j,b[2]={0};//数组b记录查找的A之前出现的P和T
    long long sum1=0,sum2=0,t;//sum1记录T的总个数，sum2记录PAT的个数
    for(i=0;a[i]!='\0';i++)
    {
        if(a[i]=='T')
            sum1++;//记录T的总个数
    }
    for(i=0;a[i]!='\0';i++)
    {
        if(a[i]=='P')
            b[0]++;//记录A之前P的个数
        else if(a[i]=='T')
            b[1]++;//记录A之前T的个数
        if(b[0]>0)//保证前面有P
        {
            if(a[i]=='A')
            {
                t=sum1-b[1];//求A之后T的个数
                sum2=sum2+b[0]*t;
            }
        }
    }
    printf("%d\n",sum2%1000000007);
}

超时原因

没记录总T的个数，然后减去A之前T的个数，用了二重循环

分析二

从末尾记录T的个数，然后记录AT的个数，最后记录PAT的个数

#include<stdio.h>
#include<string.h>
int main()
{
    char a[100005];
    scanf("%s",a);
    long long lenth=strlen(a),i,st,sat,spat;//st记录T的个数，sat记录AT的个数，spat记录PAT的个数
    sat=spat=st=0;
    for(i=lenth-1;i>=0;i--)
    {
        if(a[i]=='T')
        {
            st++;//记录T的个数
        }
        else if(a[i]=='A')
        {
            sat+=st;//记录AT的个数
        }
        else if(a[i]=='P')
        {
            spat+=sat;//记录PAT的个数
        }
    }
    printf("%lld\n",spat%1000000007);
}