1040. 有几个PAT(25)
字符串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 <cstdio>
#include <cstring>
const int maxn=100010;
const int mod=1000000007;
char str[maxn];
int leftNumP[maxn]={0};
int main(){
gets(str);
int len=strlen(str);
for(int i=0;i<len;i++){
if(i>0){
leftNumP[i]=leftNumP[i-1];
}
if(str[i]=='P'){
leftNumP[i]++;
}
}
int ans=0,rightNumT=0;
for(int i=len-1;i>=0;i--){
if(str[i]=='T'){
rightNumT++;
}else if(str[i]=='A'){
ans=(ans+leftNumP[i]*rightNumT)%mod;
}
}
printf("%d\n",ans);
return 0;
}