PAT - 乙级 1040 有几个PAT

1040. 有几个PAT（25）

时间限制

120 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CAO, Peng

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

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

输入格式：

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

输出格式：

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

输入样例：

APPAPT

输出样例：

2

思路： 以A 为中间点 左边的P的个数 乘右边的T的个数 然后所有的加起来即可

#include
#include
using namespace std;

int main(){
	string s;
	int arr[100005];
	int nump = 0, numt = 0, cnt = 0;
	getline(cin, s);
	memset(arr, 0, sizeof(arr));
	for(int i = 0; i < s.length(); i++){
		if(s[i] == 'P') nump++;      // 计算该位置之前的P的个数
		else if(s[i] == 'A'){        // 若此处=A 则将P的个数存入
			arr[i] = nump;
		}
	}
	for(int i = s.length() - 1; i >= 0; i--){
		if(s[i] == 'T') numt++;      //计算T时 从后往前计算
		else if(s[i] == 'A'){
			cnt += arr[i] * numt;
			cnt = cnt % 1000000007;
		}
	}
	printf("%d", cnt);
	return 0;
}