PAT菜鸡进化史_乙级_1040

PAT菜鸡进化史_乙级_1040

字符串 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

思路：

这题虽然代码比较短，但是还是挺有趣的
从后往前遍历一遍
遇到T了就把后面可能的T和这次的加起来
遇到A了它可能和后面的T一起组成新的AT，于是AT的可能性就是这个A后面的T和AT的和
遇到P了它可能和后面的AT一起组成新的PAT，于是PAT的可能性就是这个P后面的AT和PAT的和

Code：

#include <iostream>
#include <string>

int main(){
    using namespace std;

    string str;
    cin >> str;
    int t_num = 0, at_num = 0, pat_num = 0;
    for (int i = str.size() - 1; i >= 0; i--){
        if (str[i] == 'T')
            t_num++;
        else if (str[i] == 'A')
            at_num = (t_num + at_num) % 1000000007;
        else
            pat_num = (at_num + pat_num) % 1000000007;
    }
    cout << pat_num;

    return 0;
}