PAT (Basic Level) Practice （中文）1040 有几个PAT (25 分)

glm233

发布于 2020-09-28 10:54:00

3860

发布于 2020-09-28 10:54:00

文章被收录于专栏：glm的全栈学习之路

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

输出样例：

思路：以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;
}

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2019/10/01 ，如有侵权请联系 cloudcommunity@tencent.com 删除

编程算法

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

编程算法

登录后参与评论

0 条评论

热度