一、题目
字符串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、pa、pat数量,最后输出pat。
关键点:
- 在pat中需每次进行除余处理。不然可能数太大,造成意料之外的错误,导致测试点不通过。
源代码(C++):
#include <iostream>
using namespace std;
int main()
{
string a;
cin >> a;
int p = 0, pa = 0, pat = 0;
for (int i = 0; i < a.length(); i++) {
if (a[i] == 'P') {
p++;
} else if (a[i] == 'A') {
pa += p;
} else if (a[i] == 'T') {
pat += pa;
pat = pat % 1000000007;
}
}
cout << pat;
}