PAT甲级1093 Count PAT‘s (25分)|C++实现

最新推荐文章于 2023-06-11 01:15:08 发布

陈xLヾ

最新推荐文章于 2023-06-11 01:15:08 发布

阅读量81

点赞数 1

分类专栏： PAT Advanced

本文链接：https://blog.csdn.net/weixin_42393947/article/details/108854229

版权

PAT Advanced 专栏收录该内容

186 篇文章 1 订阅

订阅专栏

本文详细介绍了如何解决一道关于字符串中'PAT'子串计数的编程题目。通过逻辑推理，提出利用两个数组分别记录'P'和'T'的前缀和，然后遍历字符串计算'A'形成'PAT'的数量。AC代码使用C++编写，实现了高效的解决方案，并在最后对结果取模以满足特定条件。

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

一、题目描述

原题链接
在这里插入图片描述

Input Specification:

在这里插入图片描述

Output Specification:

在这里插入图片描述

Sample Input:

APPAPT

Sample Output:

二、解题思路

经过简单的逻辑推理，我们可以知道，对于某一个确定的’A’，它能形成的’PAT’的数目为：这个’A’前面’P’的个数与这个’A’后面’T’的个数的乘积，我们就可以用两个数组来记录每个位置之前’P’的数目和之后’T’的数目，最后遍历一遍，遇到’A’我们就把它能形成的’PAT’的数目加到结果上面去，最后输出即可。

三、AC代码

#include<iostream>
#include<cstdio>
#include<string>
using namespace std;
const int maxn = 100010;
int cntP[maxn] = {0}, cntT[maxn] = {0};
int main()
{
  string str;
  long long sum = 0;
  cin >> str;
  int sze = str.size();
  for(int i=0; i<sze; i++)
  {
   	if(str[i] == 'P')
    {
      if(i==0)	cntP[i] = 1;
      else	cntP[i] = cntP[i-1] + 1;
    }
    else
      if(i>0)	cntP[i] = cntP[i-1];
  }
  for(int i=sze-1; i>=0; i--)
  {
    if(str[i] == 'T')
    {
      if(i==sze-1)	cntT[i] = 1;
      else	cntT[i] = cntT[i+1]+1;
    }
    else
      if(i<sze-1)	cntT[i] = cntT[i+1];
  }
  for(int i=0; i<sze; i++)
  {
    if(str[i] == 'A')
      sum += cntT[i] * cntP[i];
  }
  sum %= 1000000007;
  printf("%d", sum);
  return 0;
}