分析:
- 根据题意,对于给定的字符串来说,能组成多少个PAT,关键在于有多少个A
- 对于每个A而言,它能组成的PAT个数是它前面的P个数和后面的T个数的乘积
- 最后把每个A能组成的PAT个数相加即可
思路:
- 将字符存入数组的同时统计T个数
- 遍历数组,统计当前出现的P和T的个数
- 当遍历到A时,将P * (T总数-T当前个数)加到和sum中
- 如果sum大于1000000007,取余
代码如下
#include <iostream>
using namespace std;
char str[100002];
int main() {
int T = 0, len = 0; //P T 为总数
char ch; scanf("%c", &ch);
while(ch != '\n') {
str[len++] = ch;
if('T' == ch) T++;
scanf("%c", &ch);
}
int cp = 0, ct = 0; //cp ct 为当前出现次数
long long sum = 0;
for(int i = 0; i < len; i++) {
if('P' == str[i]) cp++;
if('T' == str[i]) ct++;
//对现在这个字符A来说,能组成PAT的个数为前面的P个数乘以后面的T个数
if('A' == str[i]) sum += cp * (T-ct);
if(sum >= 1000000007) sum %= 1000000007;
}
printf("%lld", sum);
}