题目信息
字符串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
第一次 运行超时的算法
#include<iostream>
using namespace std;
int main(){
string s;
cin >> s;
int count=0;
for(int i=0;i<s.length();i++){
if(s[i]=='P'){
for(int j=i+1;j<s.length();j++){
if(s[j]=='A'){
for(int k=j+1;k<s.length();k++){
if(s[k]=='T') count++;
}
}
}
}
}
cout << count;
return 0;
}
分析
须注意，此题的每次计算结果都要取余
最终代码
#include<iostream>
using namespace std;
int main(){
string s;
cin >> s;
int res=0,i=0,j,pnum=0,tnum=0;
while(i<s.length()){
if(s[i]=='A'){
for(j=0;j<i;j++){
if(s[j]=='P') pnum++;
}
for(j=i+1;j<s.length();j++){
if(s[j]=='T') tnum++;
}
res=(pnum*tnum)%1000000007;
break;
}
i++;
}
i++;
while(i<s.length()&&tnum>0){
if(s[i]=='P') pnum++;
if(s[i]=='T') tnum--;
if(s[i]=='A') res=(res+pnum*tnum)%1000000007;
i++;
}
printf("%d",res);
return 0;
}
测试结果
image.png
