字符串 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
这道题有几种解法,最开始直接用暴力算的,超时是难免的 不过还能拿一些分,对算法做了优化。
解法一:先找出A的位置,然后它左边的P的个数和右边T的个数相乘就是所有用到这个A组成的PAT的个数,然后再相加即可。
#include<stdio.h>
#include<string.h>
int main()
{
int i,j,k,l,s=0,x=0,y=0;
char a[100001];
gets(a);l=strlen(a);
for(i=0;i<l;i++)
{
if(a[i]=='T')
{
x++;
}
}
for(i=0;i<l;i++)
{
if(a[i]=='P')
y++;
if(a[i]=='T')
x--;
if(a[i]=='A')
s+=x*y;
if(s>=1000000007)
s=s%1000000007;
}
printf("%d",s);
return 0;
}
解法二:从后往前算出T的个数 再算出AT的个数 再算出PAT的个数。
#include <stdio.h>
#include <string.h>
int main()
{
char str[100001];
gets(str);
int numT = 0;
int numAT = 0;
int numPAT = 0;
for(int i = strlen(str) - 1; i >= 0; --i)
{
if(str[i] == 'T')
++numT;
else if(str[i] == 'A')
numAT = (numAT + numT) % 1000000007;
else
numPAT = (numPAT + numAT) % 1000000007;
}
printf("%d", numPAT);
return 0;
}
附上最初的解法:
#include<stdio.h>
#include<string.h>
int main()
{
int i,j,k,l,m;
long long count=0;
char a[100001];
gets(a);l=strlen(a);
for(i=0;i<l;i++)
{
if(a[i]=='P')
break;
}
for(j=l-1;j>=0;j--)
{
if(a[j]=='T')
break;
}
for(i;i<j;i++)
{
if(a[i]=='P')
{
for(m=i+1;m<j;m++)
{
if(a[m]=='A')
{
for(k=m+1;k<=j;k++)
{
if(a[k]=='T')
count++;
if(count%1000000007==0)
count=0;
}
}
}
}
}
printf("%lld",count);
return 0;
}