字符串 APPAPT 中包含了两个单词 PAT,其中第一个 PAT 是第 2 位( P),第 4 位(A),第 6 位(T);第二个 PAT 是第 3 位( P),第 4 位(A),第 6 位(T)。
现给定字符串,问一共可以形成多少个 PAT?
输入格式:
输入只有一行,包含一个字符串,长度不超过10^5,只包含 P、A、T 三种字母。
输出格式:
在一行中输出给定字符串中包含多少个 PAT。由于结果可能比较大,只输出对 1000000007 取余数的结果。
输入样例:
APPAPT
输出样例:
2
暴力求解存在超时问题,看了很多大佬写的代码,学到一种思路,只需一遍循环便能解决问题
1.首先,从左往右遍历,计数P出现的个数
2.如果P的个数不为0,并且当前字符是A,那么对于这个A来说,目前能构成PA的个数就是之前的PA个数加上当前P的个数
3.如果PA的个数不为0,并且当前字符是T,那么对于这个T来说,目前能构成PAT的个数就是之前的PAT个数加上当前PA的个数
4.注意每次运算完PAT个数,整除一下1000000007,如果放在最后再整除的话很可能数据太大,又变成一个大数问题了
代码
#include<iostream>
#include<cstring>
using namespace std;
int main()
{
string s;
long long p=0,pa=0,pat=0;//记录当前p,pa,pat的个数
cin>>s;
for(int i=0;i<s.length();i++)
{
if(s[i]=='P')p++;
else if(p&&s[i]=='A')
{
pa+=p;
}
else if(pa&&s[i]=='T')
{
pat+=pa;
pat%=1000000007;
}
}
cout<<pat<<endl;
return 0;
}