PAT 1040 有几个PAT（25)

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

peanwang

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

阅读量179

点赞数 1

CC 4.0 BY-SA版权

分类专栏： PAT-B

本文链接：https://blog.youkuaiyun.com/weixin_41256413/article/details/81503115

PAT-B 专栏收录该内容

94 篇文章

订阅专栏

本文介绍了一种高效算法，用于计算给定字符串中特定子串'PAT'的出现次数。通过预计算'T'的数量并动态更新'P'和'A'之间的组合，避免了全字符串匹配，显著提高了处理大量数据的效率。

1040 有几个PAT（25）（25 分）

字符串APPAPT中包含了两个单词“PAT”，其中第一个PAT是第2位(P),第4位(A),第6位(T)；第二个PAT是第3位(P),第4位(A),第6位(T)。
现给定字符串，问一共可以形成多少个PAT？
输入格式：
输入只有一行，包含一个字符串，长度不超过 $10^5$ ，只包含P、A、T三种字母。
输出格式：
在一行中输出给定字符串中包含多少个PAT。由于结果可能比较大，只输出对1000000007取余数的结果。
输入样例：

APPAPT

输出样例：

解析

这题挺难的，我也是看别人的思路写的。
首先计算出字符串里‘T’的个数。
再遍历一次字符串。
遇到T，就把T的个数减一，这样就能保证每遇到一个A，你都会实时地知道后面T的数量。
每遇到P，countP++，保证每一次遇到一个A，你能知道前面P的数量
每遇到A，就可以进行一次结算，

完整代码：

#include<stdio.h>
#include<string.h>
int main()
{
  char input[100001];
  scanf("%s",input);
  int len = strlen(input);
  int countT=0;
  for(int i =0;i<len;i++)
    if(input[i] == 'T')
      countT++;
  int countP=0,countA=0;
  int result=0;
  for(int i=0;i<len;i++){
    if(input[i] == 'P')
      countP++;
    else if(input[i] =='T')
      countT--;
    else if(input[i] =='A')
      result = (result+countP*countT)%1000000007;
  }
  printf("%d",result);
}