PTA 乙级 1040 有几个PAT (25分) C/C++

 

 注意本题A和T是可以重复使用的，只有P不可重复使用

暴力算法很容易超时

C/C++

#include<iostream>

using namespace std;

int main()
{
    int p = 0, a = 0;
    long sum = 0;
    string str;
    cin >> str;
    for (int i = 0; i < str.length(); ++i){
        if (str[i] == 'P')      ++p;           //统计A前面的P的个数
        else if (str[i] == 'A') a += p;        //T前面所有的A前面P的个数相加
        else if (str[i] == 'T') sum += a;      //出现PAT总数
    }
    cout << sum % 1000000007;   //按题目要求取余

}