字符串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
#include <iostream>
#include <string>
#include <vector>
using namespace std;
const int MOD = 1000000007;
const int maxn = 100010; // 啊啊啊啊啊!!!!以后开数组或者容器 一定要看好题目给定的长度
//像这一题我开成了10010 而题目要求长度不超过10^5 一直导致我出现段错误,出现段错误一定要去检查数组是否越界访问以及
//像我这样 数组是否开小了
/*
对一个确定位置的A来说,以它形成的pat的个数 等于它 左边 p的个数 乘以 右边 t的个数
*/
int main(){
string s;
cin >> s;
int len = s.length();
vector<int> pnums(maxn);
//这个循环下来便能得到 每个字符它左边 P 的个数。
for(int i=0; i<len; i++){
if(i > 0){ //如果不是0号位
pnums[i] = pnums[i - 1]; //继承上一位的结果
}
if(s[i] == 'P'){ //当前位是 P
pnums[i]++; //令pnums[i] 加一
}
}
int tnums = 0, ans = 0; // tnums记录右边T的个数, ans为答案
for(int i=len - 1; i>=0; i--){
if(s[i] == 'T'){ //当前位是 T
tnums++; //右边 T的个数加一
}
else if(s[i] == 'A'){ //当前位是 A
ans = (ans + pnums[i] * tnums) % MOD; //累计乘积
}
}
cout << ans << endl;
return 0;
}