1040. 有几个PAT

字符串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

 算法:

  1. 算法开始。
  2. 读入一个字符,如果字符是换行符号则跳到第七步。
  3. 如果字符是P,countP加一。
  4. 如果字符是A,countPA等于countPA加上countP。
  5. 如果字符是T,countPAT等于countPAT加上countPA,countPAT等于countPAT除以1000000007的余数。
  6. 返回第二步。
  7. 输出countPAT。
  8. 算法结束。
该算法时间复杂度只有O(n),空间复杂度更只有O(1)。
#include 
#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;
}

你可能感兴趣的:(c语言,PAT,PAT)