PAT-1040 有几个PAT

1040 有几个PAT (25 分)

字符串 APPAPT 中包含了两个单词 PAT,其中第一个 PAT 是第 2 位(P),第 4 位(A),第 6 位(T);第二个 PAT 是第 3 位(P),第 4 位(A),第 6 位(T)。

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

输入格式:

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

输出格式:

在一行中输出给定字符串中包含多少个 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;
}

猜你喜欢

转载自blog.csdn.net/qq_42020563/article/details/84260289