题目
字符串APPAPT中包含了两个单词“PAT”,其中第一个PAT是第2位(P),第4位(A),第6位(T);第二个PAT是第3位(P),第4位(A),第6位(T)。
现给定字符串,问一共可以形成多少个PAT?
输入格式:
输入只有一行,包含一个字符串,长度不超过10^5^,只包含P、A、T三种字母。
输出格式:
在一行中输出给定字符串中包含多少个PAT。由于结果可能比较大,只输出对1000000007取余数的结果。
输入样例:
APPAPT
输出样例:
2
实现思路:此题不难,但因为运行时间的限制,想完美达标还需要灵活思考。看到之后冒出的第一个思路,也是最容易被想到的思路就是暴力破解,循环遍历,即循环找P,然后P之后的元素循环找A,然后A之后的元素循环找T,最后加出数即为结果。但考虑题目提醒,即最终测试用例可能结果较大,一测试,果然,一半运行超时。这时候就要考虑如何精简算法,提高效率。3个字母,如果直接循环找A,然后A前面的P的数量乘以A后面的T的数量就是某个A能组成PAT的数量,而且可以在循环找A的过程中顺便得到P的数量,至于T的数量,可以先得出总数,然后减去已经出现的数量。这样算法的复杂度就从N的3次方简化为2*N。
代码如下:
#include<stdio.h>
#include<string.h>
int main()
{
char szData[100001]={0};
gets(szData);
int len=strlen(szData);
long long iCountT=0,iCountP=0;//后面防止二者乘积超出int范围
long long iCnt=0;
//计算T的总数
for(int i=0;i<len;i++)
{
if(szData[i]=='T')
{
iCountT++;
}
}
for(int j=0;j<len;j++)
{
if(szData[j]=='T')//得到后面T的数量
{
iCountT--;
}
else if(szData[j]=='P')//得到前面P的数量
{
iCountP++;
}
else
{
iCnt+=iCountT*iCountP;
}
}
printf("%d",iCnt%1000000007);
return 0;
}
错误分析:
1.运行超时,即测试用例较为庞大,三重循环超时,改用思路2即可;
2.形成结果较大,超出int范围,提前指定long long类型即可。