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
题目中PAT的总数是每一位A前的P的数量与A后的T的数量相乘
在遍历过程中遇到P就累加,遇到A就加上当前位置之前的P的数量,得到当前可以组成的pa的数量,遇到T就加上pa的数量,得到当前可以组成PAT的数量。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<stack>
#include<queue>
#include<set>
#include<map>
#include<time.h>
#include<vector>
using namespace std;
int main()
{
string s;
cin>>s;
int p,pa,pat;
p = pa = pat = 0;
int len = s.size();
for(int i=0; i<len; ++i){
if(s[i]=='P')
p++;
else if(s[i]=='A'){
pa += p;
pa %= 1000000007;
}
else {
pat += pa;
pat %= 1000000007;
}
}
cout<<pat<<endl;
return 0;
}