B1040 有几个PAT

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

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

输入格式：

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

输出格式：

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

这道题暴力做法时间复杂度n³，会超时（我的水平只做出来了暴力做法）。用的算法笔记的思路，对于每个A，其左边P的个数乘以其右边T的个数即为其能组成的PAT的个数，那么设置一个数组，从左向右遍历，使其储存当前字符之前的P的个数；T同理，从后向前遍历，乘积累加即可。
特别注意几点:
1.PAT使用gets()会编译错误；
2.当要求进行取模运算时，必须每一步都要进行取模运算，否则中间过程的某一步就可能超出int的范围导致出错；

#include<iostream>
#include<string.h>
using namespace std;
char str[100010];
int main(){
	int pat[100010]={};
	cin.getline(str,100010);
	int n=strlen(str);
	int i;
	int pnum=0;
	for(i=0;i<n;i++){
		if(str[i]=='P'){
			pat[i]=++pnum;
		}else if(str[i]=='A'){
			pat[i]=pnum;
		}
	}
	int tnum=0;
	int ans=0;
	for(i=n-1;i>=0;i--){
		if(str[i]=='T'){
			tnum++;
		}else if(str[i]=='A'){
			pat[i]*=tnum;
			ans+=pat[i];
			ans%=1000000007;
		}
	}
	printf("%d\n",ans);
	return 0;
}

网上用的人比较多的一个做法，感觉不是很容易理解：

#include<stdio.h>
#include<string.h>
char str[100010];
int main(){
	scanf("%s",str);
	int p=0;
	int pa=0;
	int pat=0;
	int n=strlen(str);
	int i;
	for(i=0;i<n;i++){
		if(str[i]=='P')
			p++;
		else if(str[i]=='A'){
			pa+=p;
			pa%=1000000007;
		}else{
			pat+=pa;
			pat%=1000000007;
		}
	}
	printf("%d\n",pat);
	return 0;
}