A1093 Count PAT's(递推关系预处理)

题目描述

给一串包含PAT的字符串，求共可以形成几个PAT？
本题数目较大，直接暴力不可取，可以想到求对于任何一个A来说，以其为中心形成的PAT数目是它之前的P的个数与他之后的T的个数的乘积，因而我们可以开两个数组分别计算每个数之前的P的个数和之后的T的个数。
为了减少代码量，第二次循环求每个数之后的T的个数的过程中可直接计算答案，不用再另外开数组。

#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN = 100010;
const int MOD = 1000000007;
char str[MAXN];
int leftNumP[MAXN]={0}; //每一位左边含P的个数
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;//右边T的个数
    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;
}