题目链接
题目描述
字符串 APPAPT 中包含了两个单词 PAT,其中第一个 PAT 是第 2 位(P),第 4 位(A),第 6 位(T);第二个 PAT 是第 3 位(P),第 4 位(A),第 6 位(T)。
现给定字符串,问一共可以形成多少个 PAT?
输入格式:
输入只有一行,包含一个字符串,长度不超过10
5
,只包含 P、A、T 三种字母。
输出格式:
在一行中输出给定字符串中包含多少个 PAT。由于结果可能比较大,只输出对 1000000007 取余数的结果。
输入样例:
APPAPT
输出样例:
2
题目大意
现给定字符串,问一共可以形成多少个 PAT
解题思路
本题的思路很巧妙
组成PAT
这个字符,只能是中间字母是A
,前面是P
,后面是T
(看起来是废话是加上在后面还是有用的hhh),所以
当我们遇到A
时就可以判断当前位置的前面
有几个P
,当前位置的后面
有几个T
,然后==前面P
的数目乘以后面T
的数目就是当前A
能组成几个PAT
==,是不是听起来挺神奇的hhh,我当时第一次接触也感觉到很神奇,但是验证了一下仔细想想,就是这么个道理
题解
#include<bits/stdc++.h>
using namespace std;
int main(){
string s;
cin>>s;
int t=0; //统计当前位置之后的T有几个
int p=0; //统计当前位置之前的P有几个
long long sum=0;
for(int i=0;i<s.length();i++){
if(s[i]=='T') t++; //先判断总共有多少个T
}
for(int i=0;i<s.length();i++){
if(s[i]=='P') p++; //如果是P的话说明当前位置前面的P多一个
if(s[i]=='A'){ //如果是A的话就用前面的P乘以后面的T
sum=(sum+(p*t%1000000007))%1000000007; //题目要求,防止超限
}
if(s[i]=='T') t--; //如果是T的话就表示后面的T少一个
}
cout<<sum;
}