题目描述:
字符串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 <stdio.h>
int main(){
char str[100000];
scanf("%s", str);
int N = strlen(str);
long int i,j,k;
long int count = 0;
for(i=0; i<N; i++){
if(str[i]=='P'){
for(j=i+1; j<N; j++){
if(str[j]=='A'){
for(k=j+1; k<N; k++){
if(str[k]=='T'){
count++;
count %= 1000000007;
}
}
}
}
}
}
printf("%d\n", count);
return 0;
}
后来看来其他大牛提交的代码,学习了。
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define N 100005
int main(void)
{
char buf[N];
char * t;
int p = 0;
long long pa = 0;
long long pat = 0;
gets(buf);
t = buf;
while(*t)
{
if(*t == 'P')
{
++p;
}
else if(*t == 'A')
{
pa += p;//每遇到一个a, 则当前pa的数量即为p的数量.
}
else if(*t == 'T')
{
pat += pa;//每遇到一个t,则当前pat的数量即为pa的数量.
}
++t;
}
printf("%d\n", (int)(pat % 1000000007));
}