PTA甲级考试真题练习93——1093 Count PAT's

题目

# 思路
在遍历的过程中统计P的个数，第一次遇到A统计这个A之后的T的个数，然后相乘加到总次数中，之后遇到T就T-1，这样在遇到T就直接相乘加到总次数即可，注意取模

代码

#include <iostream>
#include<string>
using namespace std;
const int nmax = 100005;

int main()
{
	string str;
	cin >> str;
	int cnt = 0;
	int countP = 0;
	int countT = 0;
	bool first = true;
	for (int i = 0; i < str.length(); ++i) {
		if (str[i] == 'P')
			countP++;
		else if (str[i] == 'T' && !first)
			countT--;
		else if (str[i] == 'A') {
			if (first) {
				for (int j = i + 1; j < str.length(); ++j) {
					if (str[j] == 'T')
						countT++;
				}
				first = false;
			}
			cnt += countP * countT;
			cnt %= 1000000007;
		}
	}
	cout << cnt;
	return 0;
}