1040 有几个PAT (25 分)

计数字符串中PAT的数量

最新推荐文章于 2024-10-13 20:40:27 发布

原创最新推荐文章于 2024-10-13 20:40:27 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #动态规划 #贪心算法

本文介绍了一种算法，用于计算给定字符串中特定子串（如PAT）出现的次数，并提供了一个C++实现示例。该算法通过预处理字符串来高效地计算结果。

在这里插入图片描述
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

#include<iostream>
using namespace std;
int main(){
	string s;
	cin >> s;
	int len = s.length();
	int leftNumP[100005] = { 0 };
	for (int i = 0; i < len; i++) {
		if (i > 0) leftNumP[i] = leftNumP[i - 1];
		if (s[i] == 'P') leftNumP[i]++;
	}
	int ans = 0, rightNumT = 0;
	for (int j = len - 1; j >= 0; j--) {
		if (s[j] == 'T') rightNumT++;
		else if (s[j] == 'A') ans = (ans + leftNumP[j] * rightNumT) % 1000000007;
	}
	cout << ans;
	return 0;
}