1040 有几个PAT

1040 有几个PAT

/*
	1040 有几个PAT
	思路：为每一个P和T标个数，那么当循环遇到一个A的时候，
	只需要将A前面的P的个数和A后面的T的个数相乘就是这个A能去到的答案数
*/

#include<stdio.h>
#include<iostream>
#include<vector>
#include<string>

using namespace std;
const int N = 1e5 + 10;  // 100010  多取几个，免得越界
const int mod = 1e9 + 7;	// 1000000007
int P[N], T[N];

int main(void) {
	string s;
	cin >> s;
	s = '0' + s;  // 因为i要从1开始（有i-1），所以把s整体向后移动一位，第一位作0

	int ans = 0;
	for (int i = 1; i < s.size(); i++) {
		if (s[i] == 'P') { P[i] = P[i - 1] + 1; }
		else { P[i] = P[i - 1]; }
		if (s[i] == 'T') { T[i] = T[i - 1] + 1; }
		else { T[i] = T[i - 1]; }
	}

	for (int i = 1; i < s.size(); i++) {
		if (s[i] == 'A') {
			ans = (ans + P[i] * (T[s.size() - 1] - T[i])) % mod;
		}
	}
	cout << ans;
	return 0;
}