PAT-1040 有几个PAT

  • Post author:
  • Post category:其他


1040 有几个PAT (25 分)

字符串

APPAPT

中包含了两个单词

PAT

,其中第一个

PAT

是第 2 位(

P

),第 4 位(

A

),第 6 位(

T

);第二个

PAT

是第 3 位(

P

),第 4 位(

A

),第 6 位(

T

)。

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

PAT

输入格式:

输入只有一行,包含一个字符串,长度不超过10​5​​,只包含

P



A



T

三种字母。

输出格式:

在一行中输出给定字符串中包含多少个

PAT

。由于结果可能比较大,只输出对 1000000007 取余数的结果。

输入样例:

APPAPT

输出样例:

2
//思路就是从后向前查找,先找T,再找AT,再找PAT,找到一个A,则就有其后面个数为T的AT,找到一个P,就有之后个数为AT个的PAT
#include <iostream>
#include<string>
using namespace std;
int main() {
	string str;
	cin >> str;
	long long numT = 0;
	long long numAT = 0;
	long long numPAT = 0;
	for(int i = str.length() - 1; i >= 0; i--) 
	{	
		if(str[i] == 'T') 
		{
			numT++;
		} else if(str[i] == 'A') 
		{
			numAT = (numAT + numT)%1000000007;
		} else if(str[i] == 'P') 
		{
			numPAT = (numPAT + numAT)%1000000007;
		}
	}
	cout << numPAT;	
	return 0;
}