1040. 有几个PAT(25)
时间限制
120 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CAO, Peng
字符串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
第一次敲错了,超时,换种姿势过。仔细想想,看看代码就懂了。
代码:
#include <cstdio>
#include <cstring>
#include <iostream>
#define mod 1000000007
#define LL long long
using namespace std;
char s[200000];
int xin[200000];
int main(){
scanf("%s",s);
int len=strlen(s);
int t=0;
for(int i=len-1;i>=0;i--){
if(s[i]=='T')
t++;
if(s[i]=='A')
xin[i]=t;
}
LL sum=0;
LL k=0;
for(int i=len-1;i>=0;i--){
if(s[i]=='A'){
k+=xin[i];
}
if(s[i]=='P'){
sum=(sum+k)%mod;
}
}
cout<<sum<<endl;
return 0;
}