1040 有几个PAT (25 分)
字符串 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
思路:以A为中轴找它左边P出现的个数,右边T出现的个数,如果每个A都这样暴力的话会超时,很容易想到递推关系
对于每一个A来说,下一个在它右边的A的左边P的个数是上一个A左边P的个数再加上两个之间的P的个数;
同理对于每一个A来说,下一个在它右边的A的右边T的个数是上一个A右边T的个数再减去两个之间的T的个数;
OK,结束~
// luogu-judger-enable-o2
#include<bits/stdc++.h>
#include<unordered_set>
#define rg register ll
#define inf 2147483647
#define min(a,b) (a<b?a:b)
#define max(a,b) (a>b?a:b)
#define ll long long
#define maxn 300005
#define lb(x) (x&(-x))
const double eps = 1e-6;
using namespace std;
inline ll read()
{
char ch = getchar(); ll s = 0, w = 1;
while (ch < 48 || ch>57) { if (ch == '-')w = -1; ch = getchar(); }
while (ch >= 48 && ch <= 57) { s = (s << 1) + (s << 3) + (ch ^ 48); ch = getchar(); }
return s * w;
}
inline void write(ll x)
{
if (x < 0)putchar('-'), x = -x;
if (x > 9)write(x / 10);
putchar(x % 10 + 48);
}
string s;
int main()
{
cin>>s;
ll sum=0,mod=1000000007,cnt1=0,tep=0,cnt2=0;
for(rg i=0;s[i];i++)
{
if(s[i]=='A')
{
for(rg j=i+1;s[j];j++)
{
if(s[j]=='T')
cnt2++;
}
for(rg j=0;j<i;j++)
{
if(s[j]=='P')
cnt1++;
}
tep=i;
break;
}
}
for(rg i=0;s[i];i++)
{
if(s[i]=='A')
{
for(rg m=tep;m<i;m++)
{
if(s[m]=='P')cnt1++,cnt1%=mod;
if(s[m]=='T')cnt2--,cnt2%=mod;
}
//cout<<cnt1<<" "<<cnt2<<endl;
sum+=(cnt1*cnt2)%mod;
sum%=mod;
tep=i;
}
}
cout<<sum<<endl;
return 0;
}