字符串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
解题思路:
先计算P的数量。
遇到A,则PA+=P;
与到T,则PAT+=PA;
#include<iostream>
#include<string>
using namespace std;
int ans=0;
int main(){
string use;
cin>>use;
int p=0,pa=0,pat=0;
for(string::iterator i=use.begin();i!=use.end();i++){
if(*i=='P'){
p++;
p%=1000000007;
}
else if(*i=='A'){
pa+=p;
pa%=1000000007;
}
else if(*i=='T'){
pat+=pa;
pat%=1000000007;
}
}
cout<<pat%1000000007;
}