PTA乙级1040

1040

字符串 APPAPT 中包含了两个单词 PAT，其中第一个 PAT 是第 2 位§，第 4 位(A)，第 6 位(T)；第二个 PAT 是第 3 位§，第 4 位(A)，第 6 位(T)。

现给定字符串，问一共可以形成多少个 PAT？

输入格式：
输入只有一行，包含一个字符串，长度不超过10^
​5
​​ ，只包含 P、A、T 三种字母。

输出格式：

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

输入样例：

APPAPT

输出样例：

2

思路：我的思路是先得出T的总个数，然后判断遇到P的个数，之后每遇到一次T，之前T的次数就减少一个，然后遇到A就得到当前的总PAT个数

package test1;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class PTA1040 {
    public static void main(String[] args) throws IOException {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        char[] s = in.readLine().toCharArray();
        in.close();
        int countP = 0, countT = 0;
        long pat = 0;
        for (char c : s) {
            if (c == 'T')
                countT++;
        }
        for (char c : s) {
            if (c == 'P')
                countP++;
            if (c == 'T')
                countT--;
            if (c == 'A') {
                pat = pat + countP * countT;
                if (pat > 1000000007)
                    pat = pat % 1000000007;
            }
        }
        System.out.println(pat);
    }
}

运行结果：
PATPPTAPTAPATPATPATATPTATATATAAPTAPTATP
524