PTA1040 有几个PAT

PTA 1040 有几个PAT

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

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

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

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

输入样例：

APPAPT

输出样例：

2

题解：
先找出所有T 的量，然后开始循环，时刻更新已经出现的P的量，并且将出现过的的T（排在A前）减掉，再遇到A就计算T的量和P的量相乘然后累加。

思路：
1，定义sum1=0,sum2=0用于存储P,T的个数（符合要求形式）
2，利用循环语句找出字符串中的所有T
3，开始重新遍历，记录P的值，减去T的值，在出现A时，累加sum=P的数目*A的数目(sum1**sum2)
4，对sum取余并输出。
注意：
1，总次数sum在取余前的字节超出Int范围，应用long类型定义。

代码：

#include<cstdio>
#include<iostream>
using namespace std;
int main(){
    string s;       
    int sum1=0,sum2=0;
    long sum=0;       //定义所需的各种变量
    cin>>s;           //获取字符串
    for(int i=0;i<s.length();i++)    //获取字符串中全部的T的数目
        if(s[i]=='T')
            sum1++;
    for(int i=0;i<s.length();i++) {        //开始重新遍历
        if(s[i]=='P')     //获得满足条件的P的数目
            sum2++;
        else if(s[i]=='T')     //获得满足条件的T的数目
            sum1--;
        else if(s[i]=='A')        //依据A的数目累加次数
            sum+=sum1*sum2;
    }
    cout<<sum%1000000007;         //对sum取余
    return 0;
}