字符串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所对应所有的PAT再往下循环,这样就造成了臃肿以及很多不必要的遍历,而此种方案,对每一个字符的遍历都用到了,马上投入计算,效率则高。
#include<iostream>
#include<string>
using namespace std;
int main(){
string str;
int num_t=0,num_at,num_pat=0;
cin>>str;
for(int i=str.size()-1;i>=0;--i){
if(str[i]=='T') //如果是T,则记录T的个数
_t++num;
if(str[i]=='A') //如果是A,计算出A与T组成AT的个数:即当前的这个A加上这个A之后的T。如果还是A,则就是AT的的数量加上T的数量,如果又是T,则T的个数加1,再计算的时候则再加上
num_at=(num_at+num_t)%1000000007;
if(str[i]=='P') //如果是P,则计算P与AT组成的个数
num_pat=(num_at+num_pat)%1000000007;
}
cout<<num_pat;
return 0;
}
题目链接: