1040 有几个PAT

一.问题：

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

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

输入格式：

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

输出格式：

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

输入样例：

APPAPT

输出样例：

2

二.思路：

         对于每个A,它前面的P的个数和后面的T的个数的乘积就是当前可构成PAT的数目
遍历整个字符串,把每个A可以构成的PAT累加就可以得到结果了，在这过程中要记得对1000000007 取余数，不能最后才将结果对1000000007 取余数，不然最后两个测试点过不了，不知道为什么

三.代码实现：

#include<stdio.h>
#include<string.h>
int search(char str[],int length)
{
    int P = 0;//记录当前有多少个P
    int A = 0;//记录当前有多少个A
    int T = 0;//记录当前有多少个T
    int sum = 0;//记录有多少个PAT

    //第一次循环，统计出T的总个数
    for(int i = 0;i < length;i++)
    {
        if(str[i] == 'T')
        {
            T++;
        }
    }
    //第二次循环，开始统计PAT个数，并取余
    for(int i = 0;i < length;i++)
    {
        //从头开始，在没有遇到A之前，统计P的个数
        if(str[i] == 'P')
        {
            P++;
        }
        //在没有遇到A之前,如果先遇到T，则T的数量减少，遇到一个减少1
        else if(str[i] == 'T')
        {
            T--;
        }
        //遇到了A，则计算记录当前有多少个PAT
        else
        {
            sum = (sum + P*T)%1000000007;//这里尤其注意要在累加的时候取余，更新答案
        }
    }
    return sum;
}
int main()
{
    //输入一个字符串str,只包含 P、A、T 三种字母
    char str[1000001] = {'\0'};
    scanf("%s",str);
    //计算给定字符串中包含多少个 PAT
    int result = search(str,strlen(str));
    //输出结果
    printf("%d\n",result);
    return 0;
}