PAT甲级——1093 Count PAT's——分数 25

题目

---

解法1

点击查看代码

#include<iostream>
#include<vector>
using namespace std;

const int MOD = 1000000007;
// 用一个数组 preP[i] 记录从头到 i 有多少个 'P'
// 用一个数组 sufT[i] 记录从 i 到尾巴有多少个 'T'
// 遍历每一个 'A'，答案 += preP[i] × sufT[i]
int main() {
    string s;
    cin >> s;
    int n = s.size();
    
    vector<int> preP(n, 0), sufT(n, 0);

    // 预处理 preP
    for (int i = 0; i < n; ++i) {
        if (i > 0) preP[i] = preP[i-1]; // 先继承前面的数量
        if (s[i] == 'P') preP[i]++;  // 再统计当前的数量
    }

    // 预处理 sufT
    for (int i = n-1; i >= 0; --i) {
        if (i < n-1) sufT[i] = sufT[i+1];
        if (s[i] == 'T') sufT[i]++;
    }

    long long ans = 0;
    for (int i = 0; i < n; ++i) {
        if (s[i] == 'A') {
            ans = (ans + preP[i] * sufT[i]*1LL) % MOD;
        }
    }

    cout << ans << endl;
    return 0;
}