PAT B1040 有几个PAT（25）（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

 

算法思路： 

这道题的关键在于知道什么时候会形成PAT，显然在一个位置为‘A’，并且左侧的字符串中包含'p'，右侧的字符串中包含‘T',时形成字符串‘pat’,且pat的个数等于左侧的‘P’的个数乘右侧的‘T’的个数。

所以，我们在开始时先遍历一次，统计'T’的总次数，然后从前向后遍历：

如果遇到P，countP++；

如果遇到T，countT--；

如果遇到A，则总数+=countP*countT

代码展示：

#include <cstring>
#include <cstdio>
#define MAXLEN 100010
char str[MAXLEN]= {0};
int main(){
	scanf("%s",str);
	long long  count = 0;
	long long countP = 0;
	long long countT = 0;  //代表当前位置到最后的T
	for (int i=0;i<strlen(str);i++){
		if(str[i]=='T'){
			countT++;
		}
	}
	for(int i=0;i<strlen(str);i++){
		if(str[i] == 'T'){  //如果遇到T,且对T的计数不为0时，countT--;
			if(countT>0){
				countT--;
			}
		}
		if(str[i] == 'P'){  //如果遇到P则P的值++；
			countP++;
		}
		if(str[i] == 'A'){
			count+= countP * countT;
		}
	}
	printf("%lld\n",count%1000000007);
	return 0;
}