PTA 乙级——1040 有几个PAT C++实现

字符串 APPAPT 中包含了两个单词 PAT，其中第一个 PAT 是第 2 位§，第 4 位(A)，第 6 位(T)；第二个 PAT 是第 3 位§，第 4 位(A)，第 6 位(T)。
现给定字符串，问一共可以形成多少个 PAT？

输入格式：
输入只有一行，包含一个字符串，长度不超过10^​5​​，只包含 P、A、T 三种字母。

输出格式：
在一行中输出给定字符串中包含多少个 PAT。由于结果可能比较大，只输出对 1000000007 取余数的结果。

输入样例：

APPAPT

输出样例：

2

分析

这个题要求计算所有能组成"PAT"这个单词的个数，这个是个简单的计数问题，固定一个字母，然后找到所有满足位置的其他的剩余两个字母。

比如对于所有的P找到他右边的A以及这个A右边的T，加起来求和。

一开始我准备写递归的，但是如果固定字母A，变成了计算所有A左边的P和右边的T的乘积，直接用循环就可以算出来。

数组P表示当前位置左边的P的个数，数组T表示当前位置右边的T的个数。

前两个循环求数组P和数组T，最后一个循环计算乘积的和，这个循环还能缩一缩，可以但没必要，这个亚子就只有O(n)的复杂度。

最后注意取1000000007 的余。

代码

#include <iostream>
#include <string>

using namespace std;

int main()
{
	string s;
	cin >> s;
	long long n = 0;
	int P[100001] = { 0 };
	int T[100001] = { 0 };

	for (int i = 0; i < s.length(); i++)   //  统计当前位置左边的P
	{
		if (i != 0)
			P[i] = P[i - 1];
		if (s[i] == 'P')
			P[i]++;
	}

	for (int i = s.length() - 1; i>0; i--)   //  统计当前位置右边的T
	{
		T[i] = T[i + 1];
		if (s[i] == 'T')
			T[i]++;
	}

	for (int i = 0; i < s.length(); i++)   //  计算A
	{
		if (s[i] == 'A')
			n += P[i] * T[i];
	}

	cout << n % 1000000007;


}