算法学习之路|有几个PAT

最新推荐文章于 2020-03-10 22:23:14 发布

转载最新推荐文章于 2020-03-10 22:23:14 发布 · 68 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：https://yq.aliyun.com/articles/496724

字符串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
解题思路：

先计算P的数量。

遇到A，则PA+=P;

与到T，则PAT+=PA；

#include<iostream>
#include<string>
using namespace std;
int ans=0;
int main(){
  
    string use;
    cin>>use;
    int p=0,pa=0,pat=0;
    for(string::iterator i=use.begin();i!=use.end();i++){
        if(*i=='P'){
            p++;
            p%=1000000007;
        }
        
        else if(*i=='A'){
            pa+=p;
            pa%=1000000007;
        }
        else if(*i=='T'){
            pat+=pa;
            pat%=1000000007;
        }
    }
    cout<<pat%1000000007;
}