字符串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
此题的算法参考了@ice_camel的CSDN博客,他是从后向前遍历,我用从前向后遍历写了一遍。特此感谢~
#include<iostream>
#include<string>
int main()
{
using namespace std;
string s;
cin >> s;
int64_t numP = 0;
int64_t numPA = 0;
int64_t numPAT = 0;
for (int i = 0; i < s.size(); i++)
{
if (s[i] == 'P')
numP++;
else if (s[i] == 'A')
numPA += numP;
else
numPAT += numPA;
}
cout << (numPAT % 1000000007);
}