PAT (Basic Level) Practice （中文）1040 有几个PAT (25 分)
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;
}举报
腾讯云开发者
