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

二、个人理解

本题考察动态规划。主要算法为：一次遍历字符串，按顺序统计p、pa、pat数量，最后输出pat。

---

关键点：

• 在pat中需每次进行除余处理。不然可能数太大，造成意料之外的错误，导致测试点不通过。

---

源代码（C++）：

#include <iostream>
using namespace std;
int main()
{
    string a;
    cin >> a;
    int p = 0, pa = 0, pat = 0;
    for (int i = 0; i < a.length(); i++) {
        if (a[i] == 'P') {
            p++;
        } else if (a[i] == 'A') {
            pa += p;
        } else if (a[i] == 'T') {
            pat += pa;
            pat = pat % 1000000007;
        }
    }
    cout << pat;
}