1040 有几个PAT

最新推荐文章于 2024-11-02 15:25:13 发布

Dtruth

最新推荐文章于 2024-11-02 15:25:13 发布

阅读量45

点赞数

文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/m0_47464136/article/details/129519139

版权

文章讨论了一种优化算法，用于计算给定字符串中包含特定子序列PAT的数量。初始的暴力三层循环方法因效率低下导致超时，随后的改进是通过哈希表统计P、A、T的位置，然后优化计算每个A左边的P和右边的T数量，提高了计算效率。

摘要由CSDN通过智能技术生成

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

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

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

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

思考过程：
首先想到如果三层循环暴力查找肯定会超时，改进为先统计P\A\T三种字母在字符串中出现的位置，用数组存储，然后再对剩下的三个数组三层循环暴力查找对于每个P，在其后面的A\T的组合数，但是测试点2，3，4直接超时

#include <bits/stdc++.h>
using namespace std;
int main(){
	string str;
	cin>>str;
	int len=str.length();
	int hashtableP[len]={0};
	int hashtableA[len]={0};
	int hashtableT[len]={0};
	int Acnt,Pcnt,Tcnt;
	Acnt=Pcnt=Tcnt=0;
	for(int i=0;i<len;i++)
	{
		if(str[i]=='P')
			hashtableP[Pcnt++]=i;
		else if(str[i]=='A')
			hashtableA[Acnt++]=i;
		else 
			hashtableT[Tcnt++]=i;
	}
	int cnt=0;
	for(int i=0;i<Pcnt;i++)
	{
		for(int j=0;j<Acnt;j++)
		{
			for(int k=0;k<Tcnt;k++)
			{
				if(hashtableP[i]<hashtableA[j]&&hashtableA[j]<hashtableT[k])
					cnt++;
			}
		}
	} 
	cout<<cnt% 1000000007;
}

查阅资料得知，该问题本质是一个以A为基点，查找每个A左边的P和右边的T的数量，然后相乘则是该A点对应的PAT的组合数。

#include <bits/stdc++.h>
using namespace std;
int main(){
	string str;
	cin>>str;
	int len=str.length();
	int Acnt,Pcnt,Tcnt;
	Acnt=Pcnt=Tcnt=0;
	int cnt=0;
	for(int i=0;i<len;i++)
	{
		if(str[i]=='T')
			Tcnt++;
	}
	for(int j=0;j<len;j++)
	{
		if(str[j]=='P')
			Pcnt++;
		else if(str[j]=='T')
			Tcnt--;
		else
			{
			cnt+=Tcnt*Pcnt;
			cnt=cnt%1000000007;
			}
	}
	cout<<cnt;
}