1040. 有几个PAT(25)

时间:2021-09-27 18:58:36
字符串APPAPT中包含了两个单词“PAT”,其中第一个PAT是第2位(P),第4位(A),第6位(T);第二个PAT是第3位(P),第4位(A),第6位(T)。

现给定字符串,问一共可以形成多少个PAT?

输入格式:

输入只有一行,包含一个字符串,长度不超过105,只包含P、A、T三种字母。

输出格式:

在一行中输出给定字符串中包含多少个PAT。由于结果可能比较大,只输出对1000000007取余数的结果。

输入样例:
APPAPT
输出样例:
2
#include<bits/stdc++.h>
using namespace std;
#define mod 1000000007
int main(){
    string s;
    cin>>s;
    int p=0,a=0,t=0;
    for(int i=0;i<s.size();i++)
    {
        if(s[i]=='T')
        t++;    
    } 
    for(int i=0;i<s.size();i++){
         if(s[i]=='P')    //p后面的t无效 够不成pat 所以减去t
         p++;
         if(s[i]=='T')
         t--;
         if(s[i]=='A'){
             a=(a+(p*t)%mod)%mod;//关键 
         } 
    }
    cout<<a<<endl;
    return 0;
}