字符串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>
#include <string.h>
using namespace std;
int main(){
int lp=0,la=0,lt=0;
char str[100010];
scanf("%s",&str);
long count=0;
int length=strlen(str);
while(lp<length){
while(str[lp]!='P'&&lp<length){
lp++;
}
la=lp+1;
while(la<length){
while(str[la]!='A'&&la<length){
la++;
}
lt=la+1;
while(lt<length){
while(str[lt]!='T'&<<length){
lt++;
}
if(str[lt]=='T'){
count++;
count=count%1000000007;
lt++;
}
}
la++;
}
lp++;
}
printf("%ld",count);
return 0;
}
参考了别人的方法后恍然大悟,自己之前的方法每一次找都将前一次工作重复了一遍,因此十分费时,最终的代码如下:
#include <iostream>
#include <string.h>
using namespace std;
int main(){
char str[100010];
scanf("%s",&str);
int length=strlen(str);
long cT=0,cAT=0,cPAT=0;
for(int i=length-1;i>=0;i--){
if(str[i]=='T'){
cT++;
}
else if(str[i]=='A'){
cAT=cAT+cT;
}
else{
cPAT=cAT+cPAT;
cPAT=cPAT%1000000007;
}
}
printf("%ld",cPAT);
return 0;
}