有几个PAT

今天见到的一道题，写了很久却一直运行超时，很佩服标准答案。

题目

字符串APPAPT中包含了两个单词“PAT”，其中第一个PAT是第2位（P）,第4位（A）,第6位（T）；第二个PAT是第3位 （P）,第4位（A）,第6位（T）。

现给定字符串，问一共可以形成多少个PAT？

输入描述

输入只有一行，包含一个字符串，长度不超过105，只包含P、A、T三种字母。

输出描述

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

例子

APPAPT
2

思考

• 我最初的想法是先寻找字符 ‘A’ ，用到了string类的find方法，再分别计算该字符前面 P 的个数和后面 T 的个数，二者相乘即为由这个 A 组成的所有PAT。应该是o（n²）的复杂度。

for(pos = s.find('A',pos); pos != -1; pos = s.find('A',pos)){
        num_of_P = find_num(s,'P',0,pos); //计算字符串s中，0到pos范围内P的个数
        num_of_T = find_num(s,'T',pos+1,s.length()+1);
        total += num_of_P * num_of_T;
        pos++;  //pos++是为了让下次find从下一个位置开始查找
}

• 可是对于10⁵长度的字符串而言，n²还是有点难以接受了，不妨反过来想。需要统计的是 PAT 的个数，那么每当遇到一个字符 T 时，我理解为一个有穷自动机的接受状态，即一次匹配的终点。此时需要统计的便是在字符 T 前面 PA 出现的个数。同理，可以同样认为 A 是 PA 匹配的终点，便需要统计在这个 A 前面 P 的个数。问题就这样在o（n）复杂度内得到了解决。看来我还得重新看看动态规划。

#include<iostream>
#include<string>
using namespace std;
int main()
{
	string s;
	int p,pa,pat,i;
	cin >> s;
	for(i=p=pa=pat=0; i<s.length(); i++){
		if(s[i] == 'P')
			p++;
		if(s[i] == 'A')
			pa += p;
		if(s[i] == 'T')
			pat += pa;
		pat = pat % 1000000007;
	}
	cout << pat;
}