pat1093 Count PAT's

题意：给一个字符串问这里面有多少'PAT'的子串。

思路：扫一遍记录P的数量，PA数量和PAT数量。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>

using namespace std;

const int MAX_N = 100100;
const int MOD = 1000000007;
char s[MAX_N];
int len;
int cnta, cntb, cntt;

int main() {
    // freopen("in.txt", "r", stdin);
    // freopen("out.txt", "w", stdout);
    scanf("%s", s);
    len = strlen(s);
    for (int i = 0; i < len; i++) {
        if (s[i] == 'P') {
            cnta++;
            cnta %= MOD;
        } else if (s[i] == 'A') {
            cntb += cnta;
            cntb %= MOD;
        } else if (s[i] == 'T') {
            cntt += cntb;
            cntt %= MOD;
        }
    }
    printf("%d\n", cntt);
    return 0;
}