PAT算法笔记(八)————有几个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

代码：

#include"iostream"
#include "string"
using namespace std;


int main()
{
	string s;
	cin >> s;
	int numT = 0, numAT = 0, numPAT = 0;
	for (int i = s.length(); i > 0; i--)
	{
		if (s[i] == 'T')
			++numT;
		else if (s[i] == 'A')
			numAT = (numAT + numT) % 1000000007;
		else//P
		{
			numPAT = (numPAT + numAT) % 1000000007;
		}
	}
	cout << numPAT;
}

思路和陷阱：

三层循环肯定是会超时的，所以从后向前寻找T的个数，而不寻找P后面AT的个数，这样T的个数确定后，向前遇到‘A’的话，AT的个数就会确定下来，如此，PAT就会得知，时间复杂度O（n）