PAT (Basic Level) Practice 1040 有几个PAT

1040 有几个PAT

题目链接-1040 有几个PAT

解题思路
累加每个A两边P和T的个数之积即可

• 对于每个A来说，它所能构成的PAT的数目为在它之前P的数目与它之后T的数目的乘积
• 对于A之后T的数目，我们可以先预处理一下，算出整个字符串中T的数目$t$，然后后续遍历字符串的时候每遇到一个T就$t--$，就保证了遇到A时，当前$t$就是这个A之后T的数目
• 而对于A之前P的数目我们只需遍历字符串的时候每遇到一个P就$p++$记录即可，保证了遇到A时，当前$p$就是这个A之前P的数目
• 具体操作见代码

附上代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
const int M=1e9+7;
const int INF=0x3f3f3f3f;
const double PI=acos(-1.0);
typedef long long ll;
typedef pair<int,int> PII;
string s;
int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	
	cin>>s;
	ll p=0,t=0;
	for(int i=0;i<s.length();i++)
		if(s[i]=='T')
			t++;
	ll ans=0;
	for(int i=0;i<s.length();i++){
		if(s[i]=='P') p++;
		if(s[i]=='T') t--;
		if(s[i]=='A')
			ans=(ans+p*t%M)%M;
	}
	cout<<ans<<endl;
	return 0;
}