PAT (Basic Level) Practice 1040 有几个PAT

题目描述

字符串 APPAPT 中包含了两个单词 PAT，其中第一个 PAT 是第 2 位P，第 4 位(A)，第 6 位(T)；第二个 PAT 是第 3 位P，第 4 位(A)，第 6 位(T)。

现给定字符串，问一共可以形成多少个 PAT？

输入格式：

输入只有一行，包含一个字符串，长度不超过10⁵，只包含 P、A、T 三种字母。

输出格式：

在一行中输出给定字符串中包含多少个 PAT。由于结果可能比较大，只输出对 1000000007 取余数的结果。

输入样例：

APPAPT

输出样例：

2

总结

又又又又是运行超时。

第一次代码

#include<iostream>
using namespace std;
int main()
{
    string s;
    int n=0,i,j,k,m;
    cin>>s;
    for(i=s.length()-1;i>1;i--)
    {
        if(s[i]=='T')
        {
            for(j=i-1;j>0;j--)
            {
                if(s[j]=='A')
                {
                    for(k=j-1;k>=0;k--)
                    {
                        if(s[k]=='P')
                        {
                            n++;
                        }
                    }
                }
            }
        }
    }
    cout<<n;
}

我也不知道为啥我要逆向找。。。。

改进代码

#include<iostream>
#include<string>
using namespace std;
int main()
{
    string s;
    int n=0;
    long long i,j=0,k=0,m=0;//j为T个数，k为AT个数，m为PAT个数
    getline(cin,s);
    i=s.length()-1;
    while(i>=0){
        if(s[i]=='T')j++;
        else if(s[i]=='A') k+=j;
        else m+=k;
        i--;
    }
    n=m%1000000007;
    cout<<n;
}