1040 有几个PAT (25分)
字符串 APPAPT 中包含了两个单词 PAT,其中第一个 PAT 是第 2 位§,第 4 位(A),第 6 位(T);第二个 PAT 是第 3 位§,第 4 位(A),第 6 位(T)。
现给定字符串,问一共可以形成多少个 PAT?
输入格式:
输入只有一行,包含一个字符串,长度不超过10^5
,只包含 P、A、T 三种字母。
输出格式:
在一行中输出给定字符串中包含多少个 PAT。由于结果可能比较大,只输出对 1000000007 取余数的结果。
输入样例:
APPAPT
输出样例:
2
解题思路:先统计每位对应的P的数量,然后再从最后一个位置开始遍历,统计出现的T的个数,每遇到一个A,便统计P的个数与T的个数的乘积,最后取模即可
#include <iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN = 100005;
char str[MAXN];
int p[MAXN] = { 0 };
int main() {
scanf("%s",str);
int length = strlen(str);
int countT = 0;
int total = 0;
for (int i = 0; i < length; i++){
if (i>0){
p[i] = p[i - 1];
}
if (str[i] == 'P'){
p[i]++;
}
}
for (int i = length-1; i >=0; i--){
if (str[i] == 'T'){
countT++;
}
if (str[i] == 'A'){
total = (total+p[i]*countT) % 1000000007;
}
}
printf("%d\n", total);
return 0;
}
qq_23079139
发布了40 篇原创文章 · 获赞 0 · 访问量 501
私信
关注