1093. Count PAT's (25)

    线性走一遍，如果遇到P则把当前位置之前P的个数加1，遇到A则更新当前位置之前 PA 的个数，遇到 T 则统计PAT的个数

#include <iostream>
#include <cstdio>

using namespace std;

int main(){
string s;
cin >> s;

const int N = 1000000007;
long long sum = 0;
int p = 0, pa = 0;

for(auto& c : s){
if(c == 'P') p++;
else if( c == 'A') pa += p;
else if(c == 'T') sum += pa;
}

sum %= N;
cout << sum;

return 0;
}