1.题目
https://pintia.cn/problem-sets/994805342720868352/problems/994805373582557184 找出诸如APPAPT中包含多少个PAT(注意不是指“连续”)。
2.思路
若直接暴力解会超时!!
可以先算出T的个数(一层for),然后再一次for从左至右统计P和T的个数,而在这层for循环内的最后,遇到A的时候就可以计算结果:因为遍历字符串的每个A,该A前面的P的个数和该A后面的T的个数的乘积即能构成的PAT的个数,在这个for遍历累加result即可。
3.代码
#include<cstdio>
#include<stdlib.h>
#include<algorithm>
#include<string>
#include<iostream>
using namespace std;
int main(){
string s;
cin>>s;
int len=s.length(),result=0,countp=0,countt=0;
for(int i=0;i<len;i++){//此处灰常巧妙
if(s[i]=='T')
countt++;
}
for(int i=0;i<len;i++){
if(s[i]=='P') countp++;
if(s[i]=='T') countt--;
if(s[i]=='A') result=(result+(countp*countt)%1000000007)%1000000007;
}
cout<<result;
system("pause");
}