字符串 APPAPT 中包含了两个单词 PAT,其中第一个 PAT 是第 2 位(P),第 4 位(A),第 6 位(T);第二个 PAT 是第 3 位(P),第 4 位(A),第 6 位(T)。
现给定字符串,问一共可以形成多少个 PAT?
输入格式:
输入只有一行,包含一个字符串,长度不超过10^5,只包含 P、A、T 三种字母。
输出格式:
在一行中输出给定字符串中包含多少个 PAT。由于结果可能比较大,只输出对 1000000007 取余数的结果。
输入样例:
APPAPT
输出样例:
2
思路
根据分步骤乘法计数原理,我们可以这么看:找到A,然后用 左边P的个数 × 右边T的个数,即为答案。
这里我参考了网上的例子,用2次for循环完成。
- 先从左到右扫一遍,统计T的个数
再从左往右扫一遍,顺序执行以下操作:
- 当字符为P时,numberP++;
- 当字符为T时,numberT—; // 因为中间没有字符A,T的个数要减掉
- 当字符为A时,sum = (sum + numberP * numberT) % MOD;
代码
#include<cstdio>#include<cstring>const int MOD = 1000000007;int main() {char input[100001];scanf("%s", input);int countP = 0, countA = 0, countT = 0;int sum = 0;for(int i = 0; i < strlen(input); i++) {if(input[i] == 'T') {countT++;}}for(int i = 0; i < strlen(input); i++) {if(input[i] == 'P') {countP++;}if(input[i] == 'T') {countT--;}if(input[i] == 'A') {sum = (sum + countP * countT) % MOD;}}printf("%d", sum);return 0;}
