PAT 乙級 1040 有幾個PAT (25分)

1040 有幾個PAT (25分)

字元串 APPAPT 中包含了兩個單詞 PAT，其中第一個 PAT 是第 2 位§，第 4 位(A)，第 6 位(T)；第二個 PAT 是第 3 位§，第 4 位(A)，第 6 位(T)。

現給定字元串，問一共可以形成多少個 PAT？

輸入格式：

輸入隻有一行，包含一個字元串，長度不超過10

​5

​​ ，隻包含 P、A、T 三種字母。

輸出格式：

在一行中輸出給定字元串中包含多少個 PAT。由于結果可能比較大，隻輸出對 1000000007 取餘數的結果。

輸入樣例：

APPAPT

輸出樣例：

#include <iostream>
using namespace std;
int main(){
    string input;
    cin>>input;
    long long ans = 0,number_p = 0,number_t = 0;
    for (int i = 0; i < input.length(); ++i)
        if(input[i] == 'T')
            number_t++;
    for (int i = 0; i < input.length(); ++i) {
        if(input[i] == 'T')
            number_t--;
        if(input[i] == 'P')
            number_p ++;
        if(input[i] == 'A'){
            ans += number_p * number_t;
            ans %= 1000000007;
        }
    }
    cout<<ans;
    return 0;
}