Count PAT's (25) PAT甲级真题

题目分析：

由于本题字符串长度有10^5所以直接暴力是不可取的，猜测最后的算法应该是先预处理一下再走一层循环就能得到答案，所以本题的关键就在于这个预处理的过程，由于本题字符串匹配的内容的固定的PAT，所以我们可以这样想，对于一个输入的串，我们找到每个A的位置，只要知道这个A的前面有几个P，这个A的后面有几个T，就可以得到以这个A为中心的所有种数，二者相乘即可，然后如果我们能得到0~s.size()-1范围内每个A的前面有多少个P，每个A后面有多少个T，只要从头遍历一遍并且求和就能得到最终答案，由于结果可能很大需要MOD

本题代码：

#include<iostream>
#include<string>
#include<string.h>
#include<stdio.h>
using namespace std;

const int N = 100005;
const int M = 1000000007;
int p[N];
int t[N];

int main(){
    string s;
    cin>>s;
    memset(t, 0, sizeof(t));
    memset(p, 0, sizeof(p));
    int len = s.size();
    for(int i = 1; i < len; i++){
        if(s[i-1] == 'P') p[i] = p[i-1]+1;
        else p[i] = p[i-1];
    }    
    for(int i = len-2; i >= 0; i--){
        if(s[i+1] == 'T') t[i] = t[i+1]+1;
        else t[i] = t[i+1];
    }
    int ans = 0;
    for(int i = 0; i < len; i++){
        if(s[i] == 'A'){
            int temp = p[i]*t[i]%M;
            ans += temp;
            ans %= M;
        }
    }
    printf("%d\n", ans);
    return 0;
}