PAT B1040. 有几个PAT

最新推荐文章于 2022-02-14 23:30:05 发布

原创最新推荐文章于 2022-02-14 23:30:05 发布 · 341 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#PAT乙级

PAT乙级题解专栏收录该内容

56 篇文章

订阅专栏

本文介绍了一种算法，用于统计给定字符串中特定字符组合PAT的数量。通过两次遍历字符串，分别计算每个A左侧的P数量和每个A右侧的T数量，最终得出PAT组合总数。

字符串APPAPT中包含了两个单词“PAT”，其中第一个PAT是第2位(P),第4位(A),第6位(T)；第二个PAT是第3位(P),第4位(A),第6位(T)。

现给定字符串，问一共可以形成多少个PAT？

输入格式：

输入只有一行，包含一个字符串，长度不超过10⁵，只包含P、A、T三种字母。

输出格式：

在一行中输出给定字符串中包含多少个PAT。由于结果可能比较大，只输出对1000000007取余数的结果。

输入样例：

APPAPT

输出样例：

题解

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;


int main() {
	char a[100001];
	int length, count = 0, leftPnum[100001];
	long ans = 0;
	scanf("%s", a);
	length = strlen(a);
	//统计每个A左边的P个数
	for (int i = 0; i < length; i++) {
		if (a[i] == 'P')
			count++;
		else if (a[i] == 'A')
			leftPnum[i] = count;
	}
	count = 0;
	for (int i = length - 1; i >= 0; i--) {
		if (a[i] == 'T')
			count++;
		else if (a[i] == 'A') {
			ans += leftPnum[i] * count;
		}
	}
	ans = ans % 1000000007;
	printf("%ld", ans);
	return 0;
}

//对于1个A来说，它形成的PAT的个数等于左边P的个数乘以右边T的个数
//所以先从左遍历统计P个数，再从右往左统计T个数，同时可以计算总和