题目描述
字符串APPAPT中包含了两个单词“PAT”,其中第一个PAT是第2位§,第4位(A),第6位(T);第二个PAT是第3位§,第4位(A),第6位(T)。
现给定字符串,问一共可以形成多少个PAT?
输入输出
输入描述:
输入只有一行,包含一个字符串,长度不超过105,只包含P、A、T三种字母。
输出描述:
在一行中输出给定字符串中包含多少个PAT。由于结果可能比较大,只输出对1000000007取余数的结果。
输入例子:
APPAPT
输出例子:
2
解题思路
- 这道题类似于上台阶问题,要找出所有能组成pat的字母对,我们可以采用暴力破解,但是暴力的时间复杂度太高了,换一个思路。
- 我们遍历一遍可以找到所有的A,再找A的同时,把这个A前面所有的P的个数都记录下来。
- 然后再从后往前遍历一遍,统计每个A后面的T的个数,P的个数和T的个数的乘积就是我们的答案了。
- 我再做的时候,自己的编译器是可以用gets函数读入字符串的,但是提交的时候,发现编译错误,应该是有些编译器不支持gets函数,所以我就改成用cin读入了。这点在实际考试的时候也需要注意。
代码
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int maxn=100010;
int main()
{
string str;
cin>>str;
int len=str.length() ;
int p[maxn]={0};
for(int i=0;i<len;++i)
{
if(i>0){
p[i]=p[i-1];
}
if(str[i]=='P'){
p[i]++;
}
}
int ans=0,t=0;
for(int i=len-1;i>=0;--i)
{
if(str[i]=='T'){
t++;
}
else if(str[i]=='A')
{
ans=(ans+(p[i]*t))%1000000007;
}
}
printf("%d\n",ans);
return 0;
}