PAT A1093

最新推荐文章于 2022-11-26 14:27:36 发布

二狗_

最新推荐文章于 2022-11-26 14:27:36 发布

阅读量156

点赞数 1

分类专栏： PAT

本文链接：https://blog.csdn.net/qq_32115865/article/details/77342263

版权

PAT 专栏收录该内容

18 篇文章 0 订阅

订阅专栏

使用两个树状数组，其中一个保存A左侧P的个数，一个保存从一个A到下一个A之间的T的个数。

#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;

struct tarr{
	int no,times;
	tarr()=default;
	void assign(int i,int t){
		no=i,times=t;
	}
};//定义树状数组结点，no表示所在位置，times表示个数

vector<tarr> V[2];//V[0]保存每个P的左侧的P的个数，V[1]保存从一个A到下一个A之间的T的个数

int main(void){
	//freopen("in.log","r",stdin);
	char ch;
	int i=0,countP=1,countA=0;
	tarr tmp;
	while(scanf("%c",&ch)!=EOF){
		if(ch=='P'){//当输入字符为P时，将位置和P的个数累存入V[0]
			tmp.assign(i,countP);
			V[0].push_back(tmp);
			++countP;
		}
		else if(ch=='A'){//当输入字符为A时，将位置记录，此时A右侧到下一个A出现前的T个数初始为0
			tmp.assign(i,0);
			V[1].push_back(tmp);
		}
		else if(ch=='T'&&!V[1].empty()){//当输入字符为T时，V[1]出现T的个数增加
			tarr &ttmp=V[1].back();
			++ttmp.times;
		}
		++i;
	}
	long long sum=0;//用长整形，可以不用每次取余
	while(!V[0].empty()&&!V[1].empty()){
		countA+=V[1].back().times;//从右侧往左侧扫描，countA用于累记当前A右侧所有T个个数
		int pos=V[1].back().no;
		V[1].pop_back();
		while(V[0].back().no>pos&&!V[0].empty())V[0].pop_back();
		if(!V[0].empty())sum+=countA*V[0].back().times;//对于当前位置的A，能组成PAT的个数等于其左侧P的个数与右侧T的个数的积，注意调用back()时要先判断容器是否为空
	}
	printf("%lld",sum%1000000007);
	return 0;
}