字符串 APPAPT 中包含了两个单词 PAT,其中第一个 PAT 是第 2 位(P),第 4 位(A),第 6 位(T);第二个 PAT 是第 3 位(P),第 4 位(A),第 6 位(T)。
现给定字符串,问一共可以形成多少个 PAT?
输入格式:
输入只有一行,包含一个字符串,长度不超过,只包含 P、A、T 三种字母。
输出格式:
在一行中输出给定字符串中包含多少个 PAT。由于结果可能比较大,只输出对 1000000007 取余数的结果。
输入样例:
APPAPT
输出样例:
2
思路:PAT数量等于字符数组之中每个A左边的P的数量与右遍的T的数量的乘积
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
const int maxsize=100001;
const int yu=1000000007;
int leftP[maxsize]={0},rightT[maxsize]={0};
//初始化两个数组用来存储左右两边的P与T的数量
using namespace std;
int main(){
int num=0;//用来统计PAT的数量
char a[maxsize]={};
scanf("%s",a);
int n=strlen(a);//n用来存储字符数组a的长度
for(int i=0;i<n;i++){//存储每个位置左边的P的数量
if(i>0){
leftP[i]=leftP[i-1];
}
if(a[i]=='P'){
leftP[i]++;
}
}
for(int i=n-1;i>=0;i--){//统计T的数量
if(i<n-1){
rightT[i]=rightT[i+1];
}
if(a[i]=='T'){
rightT[i]++;
}
}
for(int i=0;i<n;i++){//统计PAT的数量并取余
if(a[i]=='A'){
num+=(leftP[i]*rightT[i]);
num=num%yu;
}
}
printf("%d\n",num);
return 0;
}