1040 有几个PAT (25分)
字符串 APPAPT
中包含了两个单词 PAT
,其中第一个 PAT
是第 2 位(P
),第 4 位(A
),第 6 位(T
);第二个 PAT
是第 3 位(P
),第 4 位(A
),第 6 位(T
)。
现给定字符串,问一共可以形成多少个 PAT
?
输入格式:
输入只有一行,包含一个字符串,长度不超过10510^5105,只包含 P
、A
、T
三种字母。
输出格式:
在一行中输出给定字符串中包含多少个 PAT
。由于结果可能比较大,只输出对 1000000007 取余数的结果。
输入样例:
APPAPT
输出样例:
2
超时代码
/*
* @Descripttion:
* @version:
* @Author: jimmyhuang
* @Date: 2020-04-06 22:41:19
* @LastEditors: jimmyhuang
* @LastEditTime: 2020-04-06 22:55:38
*/
#include <iostream>
using namespace std;
int main()
{
string s;
int count = 0;
cin >> s;
for (int i = 0; i < s.length(); i++)
{
//找P
if (s[i] == 'P' && s.length()-i >2)
{
//找A
for (int j = i+1; j < s.length(); j++)
{
if (s[j] == 'A'&& s.length()-i >1)
{
for (int k = j+1; k < s.length(); k++)
{
//找T
if (s[k] == 'T')
{
count++;
}
}
}
}
}
}
cout << count << endl;
return 0;
}
没搞懂,最后参考了柳神的代码
#include <iostream>
#include <string>
using namespace std;
int main() {
string s;
cin >> s;
int len = s.length(), result = 0, countp = 0, countt = 0;
for (int i = 0; i < len; i++) {
if (s[i] == 'T')
countt++;
}
for (int i = 0; i < len; i++) {
if (s[i] == 'P') countp++;
if (s[i] == 'T') countt--;
if (s[i] == 'A') result = (result + (countp * countt) % 1000000007) % 1000000007;
}
cout << result;
return 0;
}
主要思路:每次扫描到A,用这个A组成的PAT刚好是A前面的P数量*A后面T的数量