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
我一拿到题就立马写了个三重循环,不出意外超时,25分的题果然没那么简单,想了半天没想出来,在网上一个老哥的博客上看到了一个很棒的做法,马上就自己写了一个,虽然不算转载,但思路都是人家的,这里给链接:
https://www.cnblogs.com/asinlzm/p/4440603.html
#include <iostream>
#include <cstring>
using namespace std;
int main(){
char str[100000];
scanf("%s", str);
long long numP = 0, numA = 0, numT = 0;
int len = strlen(str);
while (len >= 0){
len--;
if (str[len] == 'T')
numT++;
else if (str[len] == 'A')
numA += numT;
else if (str[len] == 'P')
numP += numA;
}
printf("%d", numP % 1000000007);
return 0;
}
这里用三个变量numP、numA、numT分别表示该字符的组合次数,比如numA表示当前A能和后面出现的T组合出现的次数。
这里的思路是:逆序遍历,记录T出现的次数numT,一旦A出现,则可以知道该A可以与后面的T组成多少个“AT”序列,下一次A再出现,记录当前A的组合次数后,与先前出现的A的组合次数相加,逐个累计到numA。同理,’P’一旦出现,numA就是该P可以与其后所有能组成”AT”的序列个数,逐个累计到numP,得到的自然是所有“PAT”的个数。
果然我还差得远呢:)