字符串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
思路:PAT个数等于每个A前面P的个数乘每个A后面T的个数相加之和
#include<stdio.h>
#include<string.h>
#define N 100005
int main()
{
long long count=0;
int i=0,k=0,p=0,t=0,b[N],a[N],n;
char s[N],ch;
while((ch=getchar())!='\n')
{
if(ch=='P')p++;
if(ch=='A')a[k++]=p;
s[i++]=ch;
}
s[i]='\0';
n=k;
for(i=strlen(s)-1;i>=0;i--)
{
if(s[i]=='T')t++;
if(s[i]=='A')b[--k]=t;
}
for(i=0;i<n;i++)
{
count+=a[i]*b[i];
count%=1000000007;
}
printf("%lld",count);
}