思路:先写好输入
暴力会超时 所以想别的办法 比如可以找A 对每个A,看A前面有几个P,后面有几个T,P和T的乘积就是形成PAT的个数。
怎么获得每个A左边的P:设定一个数组LeftNumP,从左到右遍历字符串,若当前位i是P,那么leftNumP【i】就等于leftNumP[i-1]+1。如果当前位i不是P,那么leftNumP[i]=leftNumP[i-1]
获得T:直接定义一个变量numT,如果是T,加一,如果是A,直接计算结果ans,ans加上现在leftnumP和numT的乘积。这样遍历完了就能得到ans。
#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]++;
}
//这样写不对 i=0时没有i-1
// else{
// LeftNumP[i]=LeftNumP[i-1];
// }
}
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;
}