Codechef TANDEM

定义形如XXX的串是好的串,在好的串的前提下,若这个串后面的第一个字符和这个串的第一个字符不同,这个串就是有趣的串,否则是无聊的串,求S的所有子串中,有趣的串和无聊的串的个数

枚举XXX中X的长度L,每L个位置设一个关键点,那么一个好的串一定会且仅会经过3个关键点,枚举第一个,计算他和第二个关键点的LCP,LCS,可以得到好的串合法的左端点区间,注意到若一个好的串将他右移后仍然是好的串,那么他一定不是有趣的串,所以我们只要看最右边那个左端点的串是否有趣

code:

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;

const int maxn = 210000;
const int maxd = 20;

int n;
char str[maxn];
struct Suffix_Array
{
    int Fir[maxn],Sec[maxn];
    int Sa[maxn],Rank[maxn];
    int t[maxn];
    void Sort(int a[],int rk[],int sa[],int n,int m)
    {
        for(int i=0;i<=m;i++) t[i]=0;
        for(int i=1;i<=n;i++) t[a[rk[i]]]++;
        for(int i=1;i<=m;i++) t[i]+=t[i-1];
        for(int i=n;i>=1;i--) sa[t[a[rk[i]]]--]=rk[i];
    }
    void Get_Sa(int S[],int n,int m)
    {
        for(int i=1;i<=n;i++) Rank[i]=i;
        Sort(S,Rank,Sa,n,m);
        Rank[Sa[1]]=1;
        for(int i=2;i<=n;i++) Rank[Sa[i]]=Rank[Sa[i-1]]+(S[Sa[i]]!=S[Sa[i-1]]);
        int t=1;
        while(Rank[Sa[n]]!=n)
        {
            for(int i=1;i<=n;i++)
            {
                Fir[i]=Rank[i];
                Sec[i]=i+t>n?0:Rank[i+t];
                Sa[i]=i;
            }
            Sort(Sec,Sa,Rank,n,n);
            Sort(Fir,Rank,Sa,n,n);
            Rank[Sa[1]]=1;
            for(int i=2;i<=n;i++) Rank[Sa[i]]=Rank[Sa[i-1]]+
                                            (Fir[Sa[i]]!=Fir[Sa[i-1]]||Sec[Sa[i]]!=Sec[Sa[i-1]]);
            t<<=1;
        }
    }
    int Height[maxn];
    void Get_Height(int S[])
    {
        int h=0;
        for(int i=1;i<=n;i++)
        {
            if(h) h--;
            if(Rank[i]==1) continue;
            while(S[Sa[Rank[i]-1]+h]==S[i+h]) h++;
            Height[Rank[i]]=h;
        }
    }
}SA1,SA2;

struct RMQ
{
    int ln[maxn];
    int pre[maxn][maxd],suf[maxn][maxd];
    void Build(int a[])
    {
        for(int i=1;(1<<i)<=n;i++) ln[1<<i]=i;
        for(int i=1;i<=n;i++) if(!ln[i]) ln[i]=ln[i-1];
        for(int i=1;i<=n;i++) pre[i][0]=suf[i][0]=a[i];

        for(int j=1;(1<<j)<=n;j++)
        {
            for(int i=1;i+(1<<j)-1<=n;i++)
                suf[i][j]=min(suf[i][j-1],suf[i+(1<<j-1)][j-1]);
            for(int i=(1<<j);i<=n;i++)
                pre[i][j]=min(pre[i][j-1],pre[i-(1<<j-1)][j-1]);
        }
    }
    int Query(int l,int r)
    {
        if(l>r) swap(l,r); l++;
        int len=ln[r-l+1];
        return min(suf[l][len],pre[r][len]);
    }
}rmq1,rmq2;

int S1[maxn],S2[maxn];

int main()
{
    //freopen("tmp.in","r",stdin);
    //freopen("tmp.out","w",stdout);

    scanf("%s",str+1); n=strlen(str+1);
    for(int i=1;i<=n;i++) S1[i]=str[i]-'a'+1; S1[n+1]=-1;
    for(int i=1;i<=n;i++) S2[i]=str[n-i+1]-'a'+1; S2[n+1]=-1;

    SA1.Get_Sa(S1,n,26);
    SA1.Get_Height(S1);
    rmq1.Build(SA1.Height);

    SA2.Get_Sa(S2,n,26);
    SA2.Get_Height(S2);
    rmq2.Build(SA2.Height);

    ll sum=0,sum2=0;
    for(int l=1;l*3<=n;l++)
    {
        for(int i=l;i+l<=n;i+=l)
        {
            int j=i+l;
            int cp=rmq2.Query(SA2.Rank[n-i+1],SA2.Rank[n-j+1]),cs=rmq1.Query(SA1.Rank[i],SA1.Rank[j]);
            int L=max(i-l+1,i-cp+1),R=min(i,j+cs-3*l);
            if(L<=R)
            {
                sum+=R-L+1;
                if(S1[R]!=S1[R+3*l]) sum2++;
            }
        }
    }
    printf("%lld %lld\n",sum2,sum-sum2);

    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值