PAT乙级1040-有几个PAT（Java）

import java.io.*;
public class Main{
    public static void main(String args[])throws IOException{
        //BufferedReader读取速度快，保证不超时
        BufferedReader in=new BufferedReader(new InputStreamReader(System.in));
        String s=in.readLine();//读取字符串
        //分别记录每一个A前的P的数量，A后的T的数量
        int countp=0,countt=0;
        int res=0;
        //先得出T的总数
        for(int i=0;i<s.length();i++){
            if(s.charAt(i)=='T'){
                countt++;
            }
        }
        for(int i=0;i<s.length();i++){
            if(s.charAt(i)=='P'){
                countp++;
            }else if(s.charAt(i)=='T'){
                countt--;//减去A前T的数量得到A后T的数量
            }else {
                //能构成PAT的数量=每一个A前P的数量*A后T的数量
                //两次取模保证不会超出int