PAT乙 1040. 有几个PAT（25）

1040. 有几个PAT（25）

题目地址：1040. 有几个PAT（25）

题目描述：

字符串APPAPT中包含了两个单词“PAT”，其中第一个PAT是第2位(P),第4位(A),第6位(T)；第二个PAT是第3位(P),第4位(A),第6位(T)。
现给定字符串，问一共可以形成多少个PAT？

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

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

---

求解方法
如果按照常规遍历的方法，一定是会超时的，这里给出大家一个比较快速的方法：首先去计算一共有多少个T。然后我们从头开始遍历，每遇到一个P，P的计数器就+1，每遇到一个A，总和就增加P的计数器*1*T的计数器，每遇到一个T，T的计数器就-1。通过这样遍历一轮，算出的结果就是答案。

---

程序：

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

int main()
{
    char ch[100001];
    gets(ch);
    int len = strlen(ch);
    int countP = 0, countT = 0;

    for (int i = 0; i < len; i++)
    {
        if (ch[i] == 'T')
        {
            countT++;
        }
    }

    long sum = 0;
    for (int j = 0; j < len; j++)
    {
        if (ch[j] == 'P')
        {
            countP++;
        }
        else if (ch[j] == 'A')
        {
            sum += countP * countT;
        }
        else
        {
            countT--;
        }
    }

    printf("%ld\n", sum % 1000000007);
    return 0;
}