1093 Count PAT's

最新推荐文章于 2025-02-03 10:49:24 发布

原创最新推荐文章于 2025-02-03 10:49:24 发布 · 228 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#PAT甲级 #思维

算法同时被 3 个专栏收录

108 篇文章

订阅专栏

PAT甲级

98 篇文章

订阅专栏

PAT甲级部分练习题题解

80 篇文章

订阅专栏

本文介绍了一种高效算法，用于统计给定字符串中特定子串'PAT'的出现次数。通过预处理字符串，记录每个位置左侧'P'和右侧'T'的数量，实现了O(N)时间复杂度的解决方案。

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

在这里插入图片描述

题目大意：

给一个字符串输出里面有几个“PAT”。

解题思路：

暴力法O(N3)肯定超时，进一步思考我们找‘A’然后分别从左右找‘P’和‘T’可以将时间复杂度降低到O（N2），继续思考我们可以记录字符串每个位置左边有多少个‘P’，右边有多少个‘T’，然后找‘A’将时间复杂度降低至O(N)。
代码如下：

#include<iostream>
#include<cstdio>
#include<fstream>
#include<set>
#include<cmath>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<iomanip>
#include<cstdlib>
#include<list>
#include<queue>
#include<stack>
#include<algorithm>
#define inf 0x3f3f3f3f
#define MOD 1000000007
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define meminf(a) memset(a,inf,sizeof(a))
//vector ::iterator it;
//set<int>::iterator iter;
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int main()
{
//  std::ios::sync_with_stdio(false);
//  cin.tie(0);
//  freopen("test.txt","r",stdin);
//  freopen("output.txt","w",stdout);
 string s;
 int pnum[100010]={0},tnum[100010]={0};
 cin>>s;
 if(s[0]=='P')pnum[0]=1;//统计每个位置左边有多少P
 for(int i=1;i<s.length();i++)
 {
   if(s[i]=='P')pnum[i]=pnum[i-1]+1;
   else pnum[i]=pnum[i-1];
 }
 if(s[s.length()-1]=='T')tnum[s.length()-1]=1;//统计每个位置右边有多少个T
 for(int i=s.length()-2;i>=0;i--)
 {
   if(s[i]=='T')tnum[i]=tnum[i+1]+1;
   else tnum[i]=tnum[i+1];
 }
 ll len=0;
 for(int i=0;i<s.length();i++)
 {
   if(s[i]=='A')//计算PAT的数量
   {
    // cout<<pnum[i]<<' '<<tnum[i]<<endl;
     len=(len+(pnum[i]*tnum[i])%MOD)%MOD;
   }
 }
 cout<<len<<endl;
 return 0;
}