【PAT甲级】1093 Count PATs (25分)

解题过程的小记录，如有错误欢迎指出。

难度：三星（为了不超时的思想巧妙~）

题目分析

给出一串只含PAT三种字母的字符串，要求出能取出的PAT字母的个数（按顺序单个取出）

注意点

1. 结果给出的数字需要取模，可以在每次累加后取

我的解题过程

思路

1. 形成的PAT个数可以通过找出字符串中的A，对于特定的A，找出它前面所有的P的个数和它后面所有的T的个数，相乘就是所有含有这个A的PAT的个数，然后把所有A的结果累加得出
2. 正向扫描找出每个字母前面的P的数量（如果这个字母恰好是P的话则包含其本身）
3. 反向扫描找出每个字母后的T的数量（如果这个字母恰好是T的话则包含其本身），在反向扫描的同时就可以计算每个A会被包含在多少PAT中，因为所需要得知的两个因子都已知（A之前的P的个数，A之后的T的个数），同时进行累加

bug

1. 开始的时候想到了计算字母的个数然后相乘，只不过是以P为连接点，这样省时效率不高，用中间的连接点比较好
2. 输出的时候也产生了bug，习惯性的用了+=，然鹅本题这样是不行的，因为这样还是会超过给出的mod值，一定要每次加完给大括号括起来然后赋值，相当于每个结果都经过取模

代码

#include<iostream>
#include<string>
#include<vector>

using namespace std;
//啊啊啊啊啊啊啊啊啊啊就差一点啊呜呜呜
int main()
{
	string s;
	long long count = 0;
	cin >> s;
	int pcount = 0;
	vector<int> P(s.size() + 1), T(s.size() + 1), Apos(s.size() + 1);
	//正向扫描统计每个字母前(如果是P的话包括本身)有多少个P
	P[0] = 0;
	for (int i = 0; i < s.size(); i++) {
		if (i != 0) {
			P[i] = P[i - 1];
		}
		if (s[i] == 'P') P[i]++;
	}
	//反向扫描统计每个字母后(如果是T的话包括本生)有多少个T
	//为了减少循环次数直接在反向扫描的时候进行结果的累加计算
	T[s.size()] = 0;
	for (int i = s.size() - 1; i >= 0; i--) {
		T[i] = T[i + 1];
		if (s[i] == 'T') T[i]++;
		else if (s[i] == 'A') count = (count + P[i] * T[i]) % 1000000007;//一个易错点，原来习惯性写成count +=P[i] * T[i] % 1000000007;这样会超过mod的范围
	}
	cout << count;
    return 0;
}

dalao的代码

全部代码因版权原因不放出来，大家可以自行去柳神博客购买或者参考晴神的上机笔记~

借鉴点

也可以两次都正向遍历，第一遍先统计T的数量，然后再次正向时，每次遍历一个T就减去，同时统计P的数量，这样的话不用使用数组，直接用两个变量存储即可

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;
 }