1040. 有几个PAT(25)
时间限制
120 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CAO, Peng
字符串APPAPT中包含了两个单词“PAT”,其中第一个PAT是第2位(P),第4位(A),第6位(T);第二个PAT是第3位(P),第4位(A),第6位(T)。
现给定字符串,问一共可以形成多少个PAT?
输入格式:
输入只有一行,包含一个字符串,长度不超过105,只包含P、A、T三种字母。
输出格式:
在一行中输出给定字符串中包含多少个PAT。由于结果可能比较大,只输出对1000000007取余数的结果。
输入样例:APPAPT输出样例:
2
#include<iostream>
#include<cstring>
using namespace std;
int main(){
string s;
int arr[100005];
int nump = 0, numt = 0, cnt = 0;
getline(cin, s);
memset(arr, 0, sizeof(arr));
for(int i = 0; i < s.length(); i++){
if(s[i] == 'P') nump++; // 计算该位置之前的P的个数
else if(s[i] == 'A'){ // 若此处=A 则将P的个数存入
arr[i] = nump;
}
}
for(int i = s.length() - 1; i >= 0; i--){
if(s[i] == 'T') numt++; //计算T时 从后往前计算
else if(s[i] == 'A'){
cnt += arr[i] * numt;
cnt = cnt % 1000000007;
}
}
printf("%d", cnt);
return 0;
}