PAT-BASIC-1040-有几个PAT

字符串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

核心思想是统计每个字符A的左侧有多少个P,右侧有多少个T..
为了减少遍历次数,我们第一次遍历获得整个串中有多少个T
第二次从左往右遍历串,遇见P则numP++,代表下一个A之前有多少个P,遇见T则numT--,代表下一个A之后还有多少个T..
最后注意一下LL和MOD就可以了..
#include <bits/stdc++.h>
#define LL long long
#define MAXN 100000+50
#define MOD 1000000007
using namespace std;

char str[MAXN];
LL res = 0;
LL numP = 0, numT = 0;
int  main(){
    scanf("%s", str);
    int len = strlen(str);
    for(int i = 0; i < len; ++i){
        if(str[i] == 'T'){
            numT++;
        }
    }
    for(int i = 0; i < len; ++i){
        if(str[i] == 'P'){
            numP++;
        }
        else if(str[i] == 'A'){
            res = (res+numP*numT)%MOD;
        }
        else if(str[i] == 'T'){
            numT--;
        }
    }
    cout << res << endl;
    return 0;
}
CAPOUIS'CODE

 

posted @ 2015-07-16 21:04  CAPOUIS  阅读(179)  评论(0)    收藏  举报