1040. 有几个PAT(25)

时间:2022-02-25 09:48:31

字符串APPAPT中包含了两个单词“PAT”,其中第一个PAT是第2位(P),第4位(A),第6位(T);第二个PAT是第3位(P),第4位(A),第6位(T)。

现给定字符串,问一共可以形成多少个PAT?

输入格式:

输入只有一行,包含一个字符串,长度不超过105,只包含P、A、T三种字母。

输出格式:

在一行中输出给定字符串中包含多少个PAT。由于结果可能比较大,只输出对1000000007取余数的结果。

输入样例:
APPAPT
输出样例:

2

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

int main(){
	string str;
	cin>>str;
	int i,j=0,k=str.size(),sum1=0,sum2=0,sum=0;
	for(i=0;i<str.size();i++)if(str[i]=='T')sum2++;
	for(i=1;i<str.size()-1;i++){
		if(str[i]=='A'){
			for(;j<i;j++){
				if(str[j]=='P')sum1++;
				if(str[j]=='T')sum2--;
			}
			sum+=(sum1%1000000007)*(sum2%1000000007);
			sum=sum%1000000007;
		}
	}
	cout<<sum;
	return 0;
}


题目比较简单,有多种思路,个人的思路是对于每一个A,左边的P的个数乘上右边的T的个数即可,记录下整个字符串中T的数目,从左边开始遍历,每到一个A,则加上遍历过的部分中的P数目或者减去T数目即可。

比较坑爹的是题目中有一句“由于结果可能比较大,只输出对1000000007取余数的结果”,这里一开始仅仅对最终结果取余发现不对,原来是每一步都要取余才行。