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
一开始的策略是把P和T的位置定了然后找中间有多少个A,结果提交了以后运行超时了。后面看了下别人的代码,可以直接定位A,然后找A前面有多少个P,A后面有多少个T,乘一乘,结果还是不过。接着优化算法,接着超时。。。n遍以后感觉这题我现在的水平是解不出来了,以下是复制的别人的代码,真的是差距还是很明显的。
#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') //碰到T,记录该T后面T的总个数(包括这个T)
++numT;
else if(str[i] == 'A') //碰到A,记录该A后面AT组合的总个数;其中包括之前统计的A后面AT的总个数加上这个A与后面全部T组合的个数(即之前统计的T的总个数)
numAT = (numAT + numT) % 1000000007;
else { //碰到P,记录该P后面PAT组合的总个数;同理
numPAT = (numPAT + numAT) % 1000000007;
}
}
printf("%d", numPAT);
return 0;
}