C++版浙大PAT乙级1040（25分）

最新推荐文章于 2021-03-28 22:53:40 发布

原创最新推荐文章于 2021-03-28 22:53:40 发布 · 143 阅读

0 ·

CC 4.0 BY-SA版权

PAT 专栏收录该内容

101 篇文章

订阅专栏

博客提及在https://pintia.cn的相关问题中，使用三重循环会超时。作者采用先找A，算出A前P的个数和A后T的个数并相乘的方法，来得到当前A对应的单词个数。

https://pintia.cn/problem-sets/994805260223102976/problems/994805282389999616

用三重循环是肯定超时的，所以我先找A，算出A前P的个数和A后T的个数，把它们相乘，就是当前这个A对应的单词个数。喵啊～

#include<iostream>
#include<string.h>
using namespace std;

int main() {
	string str;
	cin >> str;
	long int n=0, bef=0, aft=0, len=str.size();
	long int pos = str.find('A');
	// 统计第一个A前的P和后的T个数，算出第一个A对应的n
	if(pos != str.npos){
		for(int i=0; i<pos; i++){
			if(str[i]=='P'){
				bef ++;
			}
		}
		for(int i=pos+1; i<len; i++){
			if(str[i]=='T'){
				aft ++;
			}
		}
		n = bef * aft;
	}
	// 从第一个A后开始找A
	for(int i=pos+1; i<len; i++){
		for(; str[i]!='A' && i<len; i++){
			// 还没出现下一个A的时候统计P的个数
			if(str[i]=='P'){
				bef ++;
			// 还没出现下一个A的时候统计减去T的个数
			}else if(str[i]=='T'){
				aft --;
			}
		}
		if(str[i]!='A'){
			break;
		}
		n = n + aft * bef;
	}
	cout << n%1000000007;
}