1040 有几个PAT (25分)

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;
}
posted @ 2020-04-08 11:05  青山_1997  阅读(96)  评论(0编辑  收藏  举报