PAT A1093 Count PAT's (25 分) 数学问题

    题目大意：给出一串只由 'P' 'A' 'T'三种字符组成的字符串，计算可以组成多少个''PAT''。

    如果'A'的左边有M个 'P' ， 右边有N个 'T' ，那么就能组成 M * N 个 ''PAT'' 。所以这题实际上是要遍历两次字符串，第一次遍历从左向右遍历得到每个位置上对应的在左边的'P'的个数，第二次从右向左遍历得到每个位置上对应的在右边的'T'的个数，如果遍历到'A'的话就按照  M * N 相加即可。

AC代码：

#include <bits/stdc++.h>

using namespace std;

typedef unsigned long long ull;

int main()
{
    string str;
    cin >> str;
    vector<ull> p(str.length(), 0);
    vector<ull> t(str.length(), 0);
    ull ret = 0;
    if(str[0] == 'P') p[0]++;
    if(str[str.length() - 1] == 'T') t[str.length() - 1]++;
    for (int i = 1; i < str.length(); ++i)
    {
        if(str[i] == 'P') p[i] = p[i-1] + 1;
        else p[i] = p[i-1];
    }
    for (int i = str.length() - 2; i >= 0; --i)
    {
        if(str[i] == 'T') t[i] = t[i+1] + 1;
        else t[i] = t[i+1];
    }
    for (int i = 0; i < str.length(); ++i)
    {
        if(str[i] == 'A') ret = (ret + p[i] * t[i]) % 1000000007;
    }
    cout << ret;
    return 0;
}