1040 有几个PAT (25分)

Posted 2021-03-02 tanghm

25分题中不是很难的一道，没错我就是挑软柿子捏，今天一定要把basic搞完，阿弥陀佛别拖拉了。。。

---

第一次提交后三个测试点运行超时（循环过多吧）。。。。一开始是每遇到一个A就从前往后循环看这个A前面有多少P，有多少T。

第二次提交后后两个测试点答案错误，但是不超时了！微笑 ^-

由于结果可能比较大，只输出对 1000000007取余数的结果。（题目中的说明）

不要忽略这一句，这说明了个数有可能很大，所以将sum改为longlong型后面两个测试点就过了！！

呵呵。。。。

---

算法思想就是记录每一个A前后的P、T个数，然后乘一下再加一下~

#include<iostream>
#include<string>

using namespace std;

int main() {
	string a;
	int i, j, k;
	int p[100001] = { 0 };
	int t[100001] = { 0 };
	long long int sum = 0;
	int ptemp = 0, ttemp = 0;
	getline(cin, a);
	j = 0;
	//从前往后统计每一个A前P的个数
	for (i = 0; i < a.length(); i++) {
		if (a[i] == ‘P‘) {
			ptemp++;
		}
		if (a[i] == ‘A‘) {
			p[j++] = ptemp;
		}
	}
	k = 0;
	//从后往前统计每一个人A后T的个数
	for (i = a.length() - 1; i >= 0; i--) {
		if (a[i] == ‘T‘) {
			ttemp++;
		}
		if (a[i] == ‘A‘) {
			t[k++] = ttemp;
		}
	}
	//p--123  
	//t--321
	for (i = 0; i < j; i++) {
		sum += p[i] * t[j - i - 1];
	}
	
	cout << sum % 1000000007 << endl;
	return 0;
}