PAT.B1040 有几个PAT

返回目录

样例（可复制）

APPAPT

样例输出

2

注意点

1. 本题需要使用一个技巧，记录每个位上之前出现的P的总个数p[i]，以及每位上之后的T的个数t[i]，遍历整个字符串，如果出现了字符A，则将PAT总数+=p[i]*t[i]。如果不用该技巧本题很麻烦，用暴力求解会超时

#include <bits/stdc++.h>
using namespace std;

int main(){
    string a;
    cin>>a;
    int p[100010]={0},t[100010]={0};
    long long sum=0;
    for(int i=1;i<a.size();i++){
        if(a[i-1]=='P')p[i]=p[i-1]+1;
        else p[i]=p[i-1];
    }
    for(int i=a.size()-2;i>=0;i--){
        if(a[i+1]=='T')t[i]=t[i+1]+1;
        else t[i]=t[i+1];
    }
    for(int i=1;i<a.size()-1;i++){
        if(a[i]=='A')sum+=p[i]*t[i];
    }
    cout<<sum%1000000007;
	return 0;
}