字符串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,countP加一。
- 如果字符是A,countPA等于countPA加上countP。
- 如果字符是T,countPAT等于countPAT加上countPA,countPAT等于countPAT除以1000000007的余数。
- 返回第二步。
- 输出countPAT。
- 算法结束。
该算法时间复杂度只有O(n),空间复杂度更只有O(1)。
#include <stdio.h>
#define DIV 1000000007
int main(void) {
char c;
int countP = 0, countPA = 0, countPAT = 0;
while ((c = getchar())!= '\n') {
if ('P' == c) {
countP++;
}
else if ('A' == c) {
countPA += countP;
}
else if ('T' == c) {
countPAT += countPA;
countPAT %= DIV;
}
}
printf("%d", countPAT);
return 0;
}