PAT甲级 【 A1093 Count PAT’s】

题目描述

字符串APPAPT中包含了两个单词“PAT”，其中第一个PAT是第2位P,第4位A,第6位T；第二个PAT是第3位P,第4位A,第6位T。
现给定字符串，问一共可以形成多少个PAT？

输入描述：

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

输出描述：

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

输入样例：

APPAPT

输出样例：

2

分析1

对于每一个A，求出其前面的所有P的个数以及它后面的所有T的个数，然后将其相乘，然后遍历下一个A，累加即可。可先遍历一遍字符串，统计所有T，然后再依次遍历，每遇到P，让P的个数加一。每遇到T，让其个数减一(减掉A前面无效的T),遇到A，则累积当前P和T的个数。

#include <iostream>

using namespace std;

int main() {
    int sum = 0;//累积和
    string str;
    cin >> str;
    int len = str.length();
    int P = 0, T = 0;
    for (int i = 0; i < len; ++i) {//统计T的个数
        if (str[i] == 'T')
            T++;
    }
    for (int j = 0; j < len; ++j) {//计算
        if (str[j] == 'P')
            P++;
        if (str[j] == 'T')
            T--;
        if (str[j] == 'A')
            sum = (sum + P * T) % 1000000007;
    }
    cout << sum;
    return 0;
}
 

分析2

用一个数组来记录所有的T，并初始化两个计数器countP,countT。从右往左遍历整个字符串，每遇到一个T，cnt++;遇到A时将当前cnt的值写入相应的位置。然后再从头遍历，每遇到一个P，countP++，每遇到一个A，累积countP和对应数组下标元素（代表该位置往后有多少个T）的值。

#include <iostream>


using namespace std;
const int MAX = 100001;

int main() {
    int sum = 0;
    string str;
    cin >> str;
    int numT[MAX];
    int countP = 0;//累加P
    int countT = 0;//累加T
    int len = str.length();
    for (int i = len - 1; i >= 0; --i) { //从右往左统计T的数量
        if (str[i] == 'T')
            countT++;
        if (str[i] == 'A')
            numT[i] = countT;
    }
    for (int j = 0; j < len; ++j) {//从左往右，累加P并计算
        if (str[j] == 'P')
            countP++;
        if (str[j] == 'A')
            sum = (sum + countP * numT[j]) % 1000000007;
    }
    cout << sum << endl;
    return 0;
}

注： 对于最后的取模运算，不可以在算出最终的sum的值后进行，原因是int所能保存的最大正数为2^31-1，如果先累加sum,再累加过程中有可能会发生溢出。