字符串 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个数。 先遍历一遍字符串把每一位的P(包括这一位)的左边的总P个数存入数组中。 第二遍遍历从右往左遍历,遇到T, T的个数加一, 遇到A则进行P个数和T个数的乘积。
#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <string>
using namespace std;
int main(int argc, char** argv) {
string s;
cin >> s;
int num_p[s.length()] = {0}, num_t = 0, num= 0;
if(s[0] == 'P') num_p[0] = 1;
for(int i = 1; i < s.length(); i++ )
{
if(s[i] == 'P') num_p[i] = num_p[i - 1]+ 1;//i这位是p则个数是上一位+1
else num_p[i] = num_p[i - 1];//i这位不是p则个数和上一位的p个数相等
}
for(int i = s.length() - 1; i >= 0; i--)
{
if(s[i] == 'A') num = (num + (num_t * num_p[i]) %1000000007)%1000000007;
else if (s[i] == 'T') num_t++;
}
cout << num;
return 0;
}