【PAT甲级、C++】1093 Count PAT‘s (25分)

从左开始统计P的个数，从右开始统计T的个数

然后找到在字符串中找到字符A，其所能组成的PAT串的个数就是该字符左边的P的个数*右边T的个数

# include <iostream>
# include <vector>

using namespace std;

int main()
{
    string str;
    cin >> str;
    
    int sz = str.size();
    vector<int> P(sz, 0);
    vector<int> T(sz, 0);
    
    if(sz != 0){
        P[0] = (str[0] == 'P');
        T[sz-1] = (str[sz-1] == 'T');
    }
    
    // 因为逻辑运算符的优先级比算数运算的优先级要小
    // 所以如果要将条件判断的结果放在加减乘除里，就需要用括号括起来！！！！！不然出大问题
    for(int i=1;i<sz;++i){
        P[i] = P[i-1] + (str[i] == 'P');
        T[sz-1 - i] = T[sz - i] + (str[sz-1 - i] == 'T');  
    }
    
    int cnt = 0;
    for(int i=0;i<sz;i++)
        if(str[i] == 'A')
            cnt = (cnt + P[i]*T[i]) % 1000000007;
            
    cout << cnt;

    
    return 0;
}