https://pintia.cn/problem-sets/994805260223102976/problems/994805282389999616
用三重循环是肯定超时的,所以我先找A,算出A前P的个数和A后T的个数,把它们相乘,就是当前这个A对应的单词个数。喵啊~
#include<iostream>
#include<string.h>
using namespace std;
int main() {
string str;
cin >> str;
long int n=0, bef=0, aft=0, len=str.size();
long int pos = str.find('A');
// 统计第一个A前的P和后的T个数,算出第一个A对应的n
if(pos != str.npos){
for(int i=0; i<pos; i++){
if(str[i]=='P'){
bef ++;
}
}
for(int i=pos+1; i<len; i++){
if(str[i]=='T'){
aft ++;
}
}
n = bef * aft;
}
// 从第一个A后开始找A
for(int i=pos+1; i<len; i++){
for(; str[i]!='A' && i<len; i++){
// 还没出现下一个A的时候统计P的个数
if(str[i]=='P'){
bef ++;
// 还没出现下一个A的时候统计减去T的个数
}else if(str[i]=='T'){
aft --;
}
}
if(str[i]!='A'){
break;
}
n = n + aft * bef;
}
cout << n%1000000007;
}