1040. 有几个PAT(25)
字符串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
Answer:
#include<iostream>
#include<vector>
using namespace std;
int main() {
long long count = 0;
char input[100001];
cin >> input;
int i = 0;
long long p = 0, a = 0;
while(input[i]) {
while(input[i]=='P') {
p++;
i++;
}
while(input[i]=='A') {
a+=p;
i++;
}
while(input[i]=='T') {
count += a;
i++;
}
}
cout << count%1000000007;
}
PS.
本来想要是能一下五个问题都一次通过就好了。
然而运行超时了。虽然运行超时也在意料之中吧。
所以改了算法。
好像又学到了什么。
笑。