【通过PAT复习C++与数据结构】PAT-B 1030 有几个PAT（25）

题目描述

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

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

输入输出

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

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

输入例子:
APPAPT

输出例子:
2

解题思路

• 这道题类似于上台阶问题，要找出所有能组成pat的字母对，我们可以采用暴力破解，但是暴力的时间复杂度太高了，换一个思路。
• 我们遍历一遍可以找到所有的A，再找A的同时，把这个A前面所有的P的个数都记录下来。
• 然后再从后往前遍历一遍，统计每个A后面的T的个数，P的个数和T的个数的乘积就是我们的答案了。
• 我再做的时候，自己的编译器是可以用gets函数读入字符串的，但是提交的时候，发现编译错误，应该是有些编译器不支持gets函数，所以我就改成用cin读入了。这点在实际考试的时候也需要注意。

代码

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int maxn=100010;
int main()
{
	string str;
	cin>>str;
	int len=str.length() ;
	int p[maxn]={0};
	for(int i=0;i<len;++i)
	{
		if(i>0){
			p[i]=p[i-1];
		}
		if(str[i]=='P'){
			p[i]++;
		}
	}
	int ans=0,t=0;
	for(int i=len-1;i>=0;--i)
	{
		if(str[i]=='T'){
			t++;
		}
		else if(str[i]=='A')
		{
			ans=(ans+(p[i]*t))%1000000007;
		}
	}
	printf("%d\n",ans);
	return 0;
}