PAT 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
分析
PAT的数目等于每个A之前的P的数量乘以A之后T的数量之和;
代码如下
#includeusing namespace std;int main(){ string s; getline(cin,s); /* 用long long int 防止sum过大,p[]数组中p[i]表示下标为i的A前面有p[i]个P。 sp为目前的P的数量,st表示目前T的数量 */ long long int p[s.size()]={0},sp=0,st=0,sum=0; /* 先正着遍历一遍用sp记录目前碰到P的数量,一旦遇到A 就把当前P的数量储存到p[i]中,这样每个A前的P的数量就知道了*/ for(int i=0;i =0;i--){ if(s[i]=='T') st++; if(s[i]=='A') sum+=st*p[i]; } cout< <