PAT (Basic Level) Practice 1040 有几个PAT

最新推荐文章于 2024-03-02 16:10:59 发布

Fiveneves

最新推荐文章于 2024-03-02 16:10:59 发布

阅读量109

点赞数 1

分类专栏： PTA

本文链接：https://blog.csdn.net/Fiveneves/article/details/105172670

版权

PTA 专栏收录该内容

41 篇文章 0 订阅

订阅专栏

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;
}