1040 有几个PAT (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
输出样例:
2

思路:PAT数量等于字符数组之中每个A左边的P的数量与右遍的T的数量的乘积

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
const int maxsize=100001;
const int yu=1000000007;
int leftP[maxsize]={0},rightT[maxsize]={0};
//初始化两个数组用来存储左右两边的P与T的数量 
using namespace std;

int main(){
    int num=0;//用来统计PAT的数量 
    char a[maxsize]={};
    scanf("%s",a);
    int n=strlen(a);//n用来存储字符数组a的长度 
    for(int i=0;i<n;i++){//存储每个位置左边的P的数量 
        if(i>0){
            leftP[i]=leftP[i-1];
        } 
        if(a[i]=='P'){
            leftP[i]++;
        }
    }
    for(int i=n-1;i>=0;i--){//统计T的数量 
        if(i<n-1){
            rightT[i]=rightT[i+1];
        }
        if(a[i]=='T'){
            rightT[i]++;
        }
    }
    for(int i=0;i<n;i++){//统计PAT的数量并取余 
        if(a[i]=='A'){
            num+=(leftP[i]*rightT[i]);
            num=num%yu;
        }
    }
    printf("%d\n",num);
    return 0;
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容