PAT甲级真题1093 PAT 计数(三种方法）

字符串 APPAPT 中共包含两个 PAT 作为子串。

第一个子串由第二，第四和第六个字符组成，第二个子串由第三，第四和第六个字符组成。

现在给定一个字符串，请你求出字符串中包含的 PAT 的数量。

输入格式
共一行，包含一个由大写字母 P,A,T构成的字符串。

输出格式
输出字符串中包含的 PAT 的数量。

由于结果可能很大，请你输出对 1000000007 取模后的结果。

数据范围
给定字符串的长度不超过10的5次方。
输入样例：
APPAPT
输出样例：
2

#include<bits/stdc++.h>
#define MOD 1000000007;
using namespace std;
const int N=100010;
typedef long long LL;
char s[N];
int lenp[N],lent[N];
int main(){
    scanf("%s",s+1);
    int len=strlen(s+1);//注意这边s+1
    for(int i=1;i<=len;i++){
        if(s[i]!='P')
         lenp[i]=lenp[i-1]; //lenp[i]表示左边包括它本身有多少个P
        else
         lenp[i]=lenp[i-1]+1;
    }
    for(int i=len;i>0;i--){
        if(s[i]!='T')
         lent[i]=lent[i+1]; //lent[i]表示右边包括它本身有多少个T
        else
         lent[i]=lent[i+1]+1;
    }
    LL res=0;
    for(int i=1;i<=len;i++){
        if(s[i]=='A'){
            res=(res+(lenp[i-1]*lent[i+1]))%MOD;//左边P的个数乘T的个数
        }
    }
    cout<<res<<endl;
    return 0;
}

//状态机
#include <iostream>
#include <cstring>

using namespace std;

const int N = 100010, MOD = 1e9 + 7;

int n;
char s[N], p[] = " PAT";
int f[N][4];//f[i][j]表示考虑前i个字母，状态为j的数量

int main()
{
    cin >> s + 1;
    n = strlen(s + 1);

    f[0][0] = 1;
    for (int i = 1; i <= n; i ++ )
        for (int j = 0; j <= 3; j ++ ) 
        {
            f[i][j] = f[i - 1][j];
            if (s[i] == p[j]) f[i][j] = (f[i][j] + f[i - 1][j - 1])
             % MOD;
        }

    cout << f[n][3] << endl;

    return 0;
}

对于字符串中的每一个A，它能够构成字串PA的数量是它之前的所有P的数量。
对于字符串中的每一个T，它能够构成字串PAT的数量是它之前所有字串PA的数量。
#include <iostream>
using namespace std;
const int mod = 1e9 + 7;
long long p, a, b, res;
int main()
{
    string str;
    cin >> str;
    for (auto c : str)
        if (c == 'P') p ++;
        else if (c == 'A') b += p;
        else res = (res + b) % mod;
    cout << res;
}