1093. Count PAT's (25)

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

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

0 ·

CC 4.0 BY-SA版权

文章标签：

#string

PAT 专栏收录该内容

105 篇文章

订阅专栏

本文介绍了一种算法，用于计算字符串中特定字符组合的数量。通过预处理统计字符'P'和'T'出现的位置和次数，再遍历字符串找到所有'A'字符，并计算每个'A'字符前后'P'和'T'的组合数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.这道题目比较有趣，我的思路如下：

1）首先统计countP[i]，即0~i的位置上，一共有多少个P，同理可以反向统计出countT[i],即i到n-1之间有多少个T

2）遍历string，当str[i]==‘A’时，那么ans+=countP[i-1]*countT[i+1]，当然，需要根据题目要求，对100000007取模

AC代码如下：

//#include<string>
//#include <iomanip>
#include<vector>
#include <algorithm>
//#include<stack>
#include<set>
#include<queue>
#include<map>
//#include<unordered_set>
#include<unordered_map>
//#include <sstream>
//#include "func.h"
//#include <list>
#include<stdio.h>
#include<iostream>
#include<string>
#include<memory.h>
#include<limits.h>
using namespace std;

#define maxDivide 1000000007
int main(void)
{
	string str;
	cin >> str;
	int *countP = new int[str.size()];
	int *countT = new int[str.size()];
	memset(countP, 0, sizeof(countP));
	memset(countT, 0, sizeof(countT));
	if (str[0] == 'P') countP[0] = 1;
	if (str[str.size() - 1] == 'T') countT[str.size() - 1] = 1;
	for (int i = 1; i < str.size(); i++)
	{
		if (str[i] == 'P')
			countP[i] = countP[i - 1] + 1;
		else
			countP[i] = countP[i - 1];
	}

	for (int i = str.size()-2; i >=0; i--)
	{
		if (str[i] == 'T')
			countT[i] = countT[i + 1] + 1;
		else
			countT[i] = countT[i + 1];
	}
	unsigned long long ans = 0;
	for (int i = 1; i < str.size() - 1; i++)
	{
		if (str[i] == 'A')
		{
			unsigned long long tmp = (countP[i - 1] * countT[i + 1]) % maxDivide;
			ans += tmp;
			ans = ans%maxDivide;
		}
	}

	cout << ans << endl;

	return 0;
}