字符串 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

思路:

本题直接暴力求解会出现超时,可以换个角度思考问题,对于一个确定位置的A来说,直接统计它左边P的个数和右边T的个数,乘积就是可以形成的PAT个数,然后将所有的个数相加即可,那么我们要求的最主要的就是各个位置A的所有情况。

代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
/*
本题直接使用暴力解法会超时 ????
*/
#include <cstdio>
#include <iostream>
#include <string>
#include <algorithm>

using namespace std;
int main(){
string str;
cin>>str;
int countp=0,countt=0,sumcount=0;
//遍历一遍,找到T的个数
for(int i=0;i<str.length();i++){
if(str[i]=='T')
countt++;
}
for(int i=0;i<str.length();i++){
if(str[i]=='P')
countp++;
//此时在A前面有T出现,则将T的次数减减
if(str[i]=='T')
countt--;
if(str[i]=='A')
sumcount=(sumcount+ (countp * countt) % 1000000007) % 1000000007;
}
cout<<sumcount;
return 0;
}