1040. 有几个PAT（25）（转载方法）

1040. 有几个PAT（25）

 

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

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

输入格式：

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

输出格式：

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

输入样例：

APPAPT

输出样例：

2

思路：本题解法非常巧妙，思路借鉴别人的代码

#include <iostream>
#include <string>
using namespace std;
int main() {
    string s;
    cin >> s;
    int len = s.length();
    int result = 0;
    int countp = 0;
    int countt = 0;
    for (int i = 0; i < len; i++) {
        if (s[i] == 'T')
            countt++;
    }
    for (int i = 0; i < len; i++) {
        if (s[i] == 'P') countp++;
        if (s[i] == 'T') countt--;
        if (s[i] == 'A') {
            result = (result + (countp * countt) % 1000000007) % 1000000007;
        }
    }
    cout << result;
    return 0;
}



解法二：
链接：https://www.nowcoder.com/questionTerminal/5e7d025e91ab468f909cb93d431b89c3?toCommentId=22044
来源：牛客网
#include <iostream>
#include <cstdio>
#include <cstdlib>
using namespace std;
 
int main()
{
    int p=0, pa=0, pat=0;
    string str;
    getline(cin,str);
    for(size_t i=0;i<str.length();i++)
    {
        if(str[i]=='P')
        {
            p++;
        }
        else if(str[i]=='A')
        {
            pa+=p;
            pa%=1000000007;
        }
        else if(str[i]=='T')
        {
            pat+=pa;
            pat%=1000000007;
        }
    }
    printf("%d\n", pat);
    return 0;
}