【Acwing T93】PAT计数

题目链接T1583

分析以及解答

本题题意很简单，对于给定的由P、A、T构成的字符串，对PAT顺序的子串进行计数。
经过观察可以发现，PAT子串的数量实际上等于T前面PA子串的数量；PA子串的数量等于A前面P的数量。
可以考虑利用类dp的思想解题。开三个数组（也可以是一个二维的数组），分别记录P的数量，PA的数量，以及PAT的数量。
具体转移方式在代码中

#include<bits/stdc++.h>
using namespace std;
int sumA[100005];
int sumP[100005];
int sumT[100005];
const int M = 1000000007;
int main()
{
    string s;
    cin >> s;
    //cout << s.length() << ' ' << s[1];
    int a,p,t;
    a = p = t = 0;
    for(int i = 0;i < s.length();++i){
        if(s[i] == 'P'){
            sumP[i] = (sumP[p] + 1) % M;//P的数量
            p = i;
        }
        else if(s[i] == 'A'){
            sumA[i] = (sumA[a] + sumP[p]) % M;//PA的数量
            a = i;
        }
        else if(s[i] == 'T'){
            sumT[i] = (sumA[a] + sumT[t]) % M;//PAT的数量
            t = i;
        }
    }
//    cout << a << p << t;

    cout << sumT[t];
}