题目:
字符串 APPAPT 中包含了两个单词 PAT,其中第一个 PAT 是第 2 位§,第 4 位(A),第 6 位(T);第二个 PAT 是第 3 位( P),第 4 位(A),第 6 位(T)。
现给定字符串,问一共可以形成多少个 PAT?
输入格式:
输入只有一行,包含一个字符串,长度不超过105,只包含 P、A、T 三种字母。
输出格式:
在一行中输出给定字符串中包含多少个 PAT。由于结果可能比较大,只输出对 1000000007 取余数的结果。
输入样例:
APPAPT
输出样例:
2
代码长度限制
16 KB
时间限制
150 ms
内存限制
64 MB
我的第一次代码是暴力解题,用三层循加计数暴力解题,毫无疑问超时了,后来思考了一下如何优化一下算法,可以用一次循环搞定。
优化思路:循环遍历时只要记住PA的个数,遇到T时加上PA的个数就可以优化许多,用两个数分别记下P的个数,遇到了A后就加上它前面的P的个数,这样PA的个数就可以得到。
具体代码如下:
#include<stdio.h>
#include<string.h>
char a[100001];
int main()
{
gets(a);
int l=strlen(a),i,j,k,m=0,n=0,sum=0;
long long d=0;
for(register int i=0;i<l;i++)
{
if(a[i]=='P')
{
m++;
}else{
if(a[i]=='A')
{
n=1;
sum+=m;
}
if(a[i]=='T')
{
d+=sum;
}
}
}
printf("%lld",d%1000000007);
}