1040 有几个PAT

关键

• PAT的个数就是A左边P的个数乘以右边T的个数

• 时间复杂度必须为o(n)

思路

从头开始向后遍历字符串。

如果是’P'，A左边的p的count++；

如果是‘A'，cnt+=A左边的p的count*（t的count-A左边的t的count）；

如果是’T‘，A左边的t的count++

代码

#include <iostream>
#include <cstdio>
#include <string>
using namespace std;
​
​
int main(){
​
    long long cnt=0;
    long long t_cnt=0;
    long long l_p_cnt=0;
    long long l_t_cnt=0;
    string s;
    cin>>s;
​
    for(int i=0;s[i]!=NULL;i++){//统计t的总数 
        if(s[i]=='T'){
            t_cnt++;
        }
    }
    for(int i=0;s[i]!=NULL;i++){
        if(s[i]=='P'){
            l_p_cnt++;
        }
        if(s[i]=='A'){
            cnt+=l_p_cnt*(t_cnt-l_t_cnt);
        }
        if(s[i]=='T'){
            l_t_cnt++;
        }
    }
    cout<<(cnt%1000000007);
    return 0;
}

参考

(30条消息) 1040. 有几个PAT（25）- PAT乙级真题_柳婼的博客-CSDN博客