PAT 乙级 1040 有几个PAT

1040 有几个PAT （25 point(s)）

字符串 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

经验总结：

 这一题用到的是递推的思想，先遍历一遍字符串，利用两个数组p[maxn],t[maxn]，分别计算当前位置之前的P的数量（即p[i]）以及当前位置之后T的数量（即t[i]），然后再遍历一遍字符串，当前位置为A时，累加p[i]*t[i]，注意相乘结果可能超出整型，所以最好定义结果为long long int型，就这些啦~(●´▽｀●)
（还有一个常识，大的数组（大约10^5的整型就需要了）要定义在main函数外哈，不然会爆栈）

AC代码

 

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