PAT-B1040 有几个PAT（25）

1040 有几个PAT（25）（25 分）

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

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

输入格式：

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

输出格式：

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

输入样例：

APPAPT

输出样例：

2

分析：

若采用三重循环肯定会超时。对于字符串中的每个A，如果能得到这个A前面的P的数量p和这个A后面的T的数量t,则使用当前的A可以组成p*t个PAT，将所有的A能组成的PAT的数量累加起来，即可得到总的数量。

代码：

#include <cstdio>
#include <cstring>

using namespace std;

int main()
{
    char input[100100];
    int i;
    long long sum = 0;

    scanf("%s", input);
    //统计字符串中T的数量
    int t_num = 0;
    for(i = 0; i < strlen(input); i++)
        if(input[i] == 'T')
            t_num ++;
    int pba = 0, tba = 0;//记录当前的A前面的P和T的数量
    for(i = 0; i < strlen(input) - 1; i++)
    {
        if(input[i] == 'P')
            pba++;
        if(input[i] == 'T')
            tba++;
        if(input[i] == 'A')
            //将当前A所能形成的PAT的数量累加起来
            sum += pba * (t_num - tba);
    }
    printf("%lld", sum % 1000000007);
    return 0;
}