PAT乙级1040 有几个PAT(C语言)

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

分析一

单独用一次循环,统计T的个数,在用循环查找A,同时记录A之前P的个数,减去A之前出现的T,PAT个的个数即为两数乘积

代码一

#include<stdio.h>
int main()
{
    
    
    char a[100005];
    scanf("%s",a);//读取字符串
    int i,j,b[2]={
    
    0};//数组b记录查找的A之前出现的P和T
    long long sum1=0,sum2=0,t;//sum1记录T的总个数,sum2记录PAT的个数
    for(i=0;a[i]!='\0';i++)
    {
    
    
        if(a[i]=='T')
            sum1++;//记录T的总个数
    }
    for(i=0;a[i]!='\0';i++)
    {
    
    
        if(a[i]=='P')
            b[0]++;//记录A之前P的个数
        else if(a[i]=='T')
            b[1]++;//记录A之前T的个数
        if(b[0]>0)//保证前面有P
        {
    
    
            if(a[i]=='A')
            {
    
    
                t=sum1-b[1];//求A之后T的个数
                sum2=sum2+b[0]*t;
            }
        }
    }
    printf("%d\n",sum2%1000000007);
}

超时原因

没记录总T的个数,然后减去A之前T的个数,用了二重循环

分析二

从末尾记录T的个数,然后记录AT的个数,最后记录PAT的个数

#include<stdio.h>
#include<string.h>
int main()
{
    
    
    char a[100005];
    scanf("%s",a);
    long long lenth=strlen(a),i,st,sat,spat;//st记录T的个数,sat记录AT的个数,spat记录PAT的个数
    sat=spat=st=0;
    for(i=lenth-1;i>=0;i--)
    {
    
    
        if(a[i]=='T')
        {
    
    
            st++;//记录T的个数
        }
        else if(a[i]=='A')
        {
    
    
            sat+=st;//记录AT的个数
        }
        else if(a[i]=='P')
        {
    
    
            spat+=sat;//记录PAT的个数
        }
    }
    printf("%lld\n",spat%1000000007);
}

猜你喜欢

转载自blog.csdn.net/qq_51517771/article/details/114099113
今日推荐