PAT乙级真题 1040. 有几个PAT（25）

题目链接

 点击打开链接

解题思路

若用三层循环的方式，会超时。其实我们只需要找到每一个A，然后计算出这个A前面有多少个P，后面有多少个T，然后把这个两个数字相乘。最后把每一个A对应的PAT数目相加就得到了全部的pat数目。

代码

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
int main(){
	char PAT[100001];
	scanf("%s",PAT);
	int i, j, k, len,count,p_num,t_all_num,t_num;
	len = strlen(PAT);
	for (i = 0,t_all_num = 0,p_num = 0,t_num = 0; i < len;i++){
		if (PAT[i] == 'T')
			t_all_num++;
	}
	for (count = 0,i = 0; i < len ;i++){
		if (PAT[i] == 'P')
			p_num++;
		else if (PAT[i] == 'T')
			t_num++;
		else{
			count += p_num*(t_all_num - t_num);
			count %= 1000000007;
		}
	}
	printf("%d",count);

	return 0;
}