1040 有几个PAT (25分)

1040 有几个PAT (25分)

• PAT (Basic Level) Practice

---

问题：

字符串 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

---

思路：

• 柳神是我的向往的目标！
• 这道题有点找规律的感觉 ，能组成 pat 的个数恰好等于 a 之前的 p 个数和 a 之后 t 的个数的乘积

---

#include<iostream>
#include<stdio.h>
#include<set>
#include<algorithm>
#include<cmath>
#include<map>
#include<vector>
#include<string.h>
#include<stack>
#include<cctype>
using namespace std;
#define inf 0x3f3f3f3f
typedef long long ll;
int cntP=0,cntT=0,ans=0;
string s;

int main()
{  
    getline(cin,s);
    for(int i=0;i<s.size();i++) if(s[i]=='T') cntT++;
    for(int i=0;i<s.size();i++)
    {  
        if(s[i]=='P') cntP++;
        if(s[i]=='T') cntT--; 
        if(s[i]=='A') ans=(ans+cntP*cntT%1000000007)%1000000007;
    }

    printf("%d",ans);

    return 0;
}