字符串 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字符所能组成的PAT数量 = 该A左侧的P数量 X 该A右侧的T数量;总的PAT数量等于所有A所能组成的PAT数量之和
#include <iostream>
#include<map>
const int MOD = 1000000007;
using namespace std;
int main(){
string str;
cin >> str;
int pnum = 0,tnum = 0,pat = 0;
for (int i = 0; i < str.length(); i++) {
if (str[i] == 'T') {
tnum++;
}
}
for (int j = 0; j < str.length(); j++) {
if (str[j] == 'P') {
pnum++;
}
if (str[j] == 'A') {
pat = (pat + (pnum * tnum) % MOD) % MOD;
}
if (str[j] == 'T') {
tnum--;
}
}
cout << pat << endl;
return 0;
}