【后缀数组】[NOI2016]优秀的拆分_noi2016】优秀的拆分-CSDN博客

本文链接：https://blog.csdn.net/outer_form/article/details/52079558

题目描述

如果一个字符串可以被拆分为 $AABB$ 的形式，其中 $A$ 和 $B$ 是任意非空字符串，则我们称该字符串的这种拆分是优秀的。

例如，对于字符串 aabaabaa，如果令 $A = \mathrm{aab}$ ， $B = \mathrm{a}$ ，我们就找到了这个字符串拆分成 $AABB$ 的一种方式。

一个字符串可能没有优秀的拆分，也可能存在不止一种优秀的拆分。比如我们令 $A=\mathrm{a}$ ， $B=\mathrm{baa}$ ，也可以用 $AABB$ 表示出上述字符串；但是，字符串 abaabaa 就没有优秀的拆分。

现在给出一个长度为 $n$ 的字符串 $S$ ，我们需要求出，在它所有子串的所有拆分方式中，优秀拆分的总个数。这里的子串是指字符串中连续的一段。

以下事项需要注意：

出现在不同位置的相同子串，我们认为是不同的子串，它们的优秀拆分均会被记入答案。
在一个拆分中，允许出现 $A=B$ 。例如 cccc 存在拆分 $A=B=\mathtt{c}$ 。
字符串本身也是它的一个子串。

输入格式

每个输入文件包含多组数据。输入文件的第一行只有一个整数 $T$ ，表示数据的组数。保证 $1 \le T \le 10$ 。

接下来 $T$ 行，每行包含一个仅由英文小写字母构成的字符串 $S$ ，意义如题所述。

输出格式

输出 $T$ 行，每行包含一个整数，表示字符串 $S$ 所有子串的所有拆分中，总共有多少个是优秀的拆分。

样例一

input

4
aabbbb
cccccc
aabaabaabaa
bbaabaababaaba

output

explanation

我们用 $S[i, j]$ 表示字符串 $S$ 第 $i$ 个字符到第 $j$ 个字符的子串（从 $1$ 开始计数）。

第一组数据中，共有 $3$ 个子串存在优秀的拆分：

$S[1,4] = \mathtt{aabb}$ ，优秀的拆分为 $A=\mathtt{a}$ ， $B = \mathtt{b}$ ；

$S[3,6] = \mathtt{bbbb}$ ，优秀的拆分为 $A=\mathtt{b}$ ， $B = \mathtt{b}$ ；

$S[1,6] = \mathtt{aabbbb}$ ，优秀的拆分为 $A= \mathtt{a}$ ， $B = \mathtt{bb}$ 。

而剩下的子串不存在优秀的拆分，所以第一组数据的答案是 $3$ 。

第二组数据中，有两类，总共 $4$ 个子串存在优秀的拆分：

对于子串 $S[1,4] = S[2,5] = S[3,6] = \mathrm{cccc}$ ，它们优秀的拆分相同，均为 $A = \mathrm{c}$ ， $B = \mathrm{c}$ ，但由于这些子串位置不同，因此要计算 $3$ 次；

对于子串 $S[1,6] = \mathrm{cccccc}$ ，它优秀的拆分有 $2$ 种： $A = \mathrm{c}$ ， $B = \mathrm{cc}$ 和 $A = \mathrm{cc}$ ， $B = \mathrm{c}$ ，它们是相同子串的不同拆分，也都要计入答案。

所以第二组数据的答案是 $3 + 2 = 5$ 。

第三组数据中， $S[1,8]$ 和 $S[4,11]$ 各有 $2$ 种优秀的拆分，其中 $S[1,8]$ 是问题描述中的例子，所以答案是 $2 + 2 = 4$ 。

第四组数据中， $S[1,4]$ ， $S[6,11]$ ， $S[7,12]$ ， $S[2,11]$ ， $S[1,8]$ 各有 $1$ 种优秀的拆分， $S[3,14]$ 有 $2$ 种优秀的拆分，所以答案是 $5 + 2 = 7$ 。

样例二

见样例数据下载。

样例三

见样例数据下载。

限制与约定

对于全部的测试点,保证 $1 \le T \le 10$ 。以下对数据的限制均是对于单组输入数据而言的，也就是说同一个测试点下的 $T$ 组数据均满足限制条件。

我们假定 $n$ 为字符串 $S$ 的长度，每个测试点的详细数据范围见下表：

测试点编号 $n$ 其他约束 1、2 $\leq 300$ $S$ 中所有字符全部相同

3、4

≤2000 $\leq 2000$ 5、6

≤10 $\leq 10$ 无7、8

≤20 $\leq 20$ 9、10

≤30 $\leq 30$ 11、12

≤50 $\leq 50$ 13、14

≤100 $\leq 100$ 15

≤200 $\leq 200$ 16

≤300 $\leq 300$ 17

≤500 $\leq 500$ 18

≤1000 $\leq 1000$ 19

≤2000 $\leq 2000$ 20

≤30000 $\leq 30000$

时间限制： $1.5\texttt{s}$

空间限制： $512\texttt{MB}$

下载

样例数据下载

分析

CCF送80分，然后剩下的各种算法只相差5分。
直接来看满分算法。

我们将 $AA$ 和 $BB$ 分开看，计算一个字符开始有多少个 $AA$ 。考场上我 $SB$ 了，考完才想到可以像【后缀数组】[UVA10829]L-Gap substring 那样做。
枚举 $A$ 的长度 $l$ ，然后把整个字符串按照长度 $l$ 分段，计算 $s_{i*l}$ 和 $s_{(i+1)*l}$ 的 $LCP$ 和 $LCS$ ，如果长度加起来 $\ge l$ ，就说明这里有 $AA$ ，在相应位置计数器增加1，即 $cnt_{i*l-LCS+1},cnt_{i*l-LCS+1}\dots cnt_{i*l-LCS+1+(LCS+LCP-1-l)}$ 都增加1。
然后再做一遍，这次不增加计数器了，增加可能的末尾位置的计数器的和，即 $ans+=\sum_{j=(i+1)*l+LCP-1-(LCS+LCP-l)}^{(i+1)*l+LCP-1}cnt_j$ 。

代码

#include<cstdio>
#include<algorithm>
#include<cstring>
#define MAXN 300000
#define MAXLOG 19
#define MAXC 256
using namespace std;
template<class T>
void Read(T &x){
    static char c;
    bool f(0);
    while(c=getchar(),c!=EOF){
        if(c=='-')
            f=1;
        if(c>='0'&&c<='9'){
            x=c-'0';
            while(c=getchar(),c>='0'&&c<='9')
                x=x*10+c-'0';
            ungetc(c,stdin);
            if(f)
                x=-x;
            return;
        }
    }
}
int T,n,array[4][MAXN*2+10],*sa,*nsa,*rk,*nrk,b[MAXN*2+10],hn,m,st[MAXN*2+10][MAXLOG+1],height[MAXN*2+10],Log,cnt[MAXN+10],lg2[MAXN*2+10];
char s[MAXN+10];
long long ans;
void Get_sa(){
    int i,k;
    sa=array[0],nsa=array[1],rk=array[2],nrk=array[3];
    rk[n]=nrk[n]=-1;
    for(i=0;i<=MAXC;i++)
        b[i]=0;
    for(i=0;i<n;i++)
        b[s[i]]++;
    for(i=1;i<=MAXC;i++)
        b[i]+=b[i-1];
    for(i=0;i<n;i++)
        sa[--b[s[i]]]=i;
    for(rk[sa[0]]=0,i=1;i<n;i++){
        rk[sa[i]]=rk[sa[i-1]];
        if(s[sa[i]]!=s[sa[i-1]])
            rk[sa[i]]++;
    }
    for(k=1;rk[sa[n-1]]<n-1;k<<=1){
        for(i=0;i<n;i++)
            b[rk[sa[i]]]=i;
        for(i=n-1;i>=0;i--)
            if(sa[i]>=k)
                nsa[b[rk[sa[i]-k]]--]=sa[i]-k;
        for(i=n-k;i<n;i++)
            nsa[b[rk[i]]--]=i;
        for(nrk[nsa[0]]=0,i=1;i<n;i++){
            nrk[nsa[i]]=nrk[nsa[i-1]];
            if(rk[nsa[i-1]]!=rk[nsa[i]]||rk[nsa[i-1]+k]!=rk[nsa[i]+k])
                nrk[nsa[i]]++;
        }
        swap(rk,nrk);
        swap(sa,nsa);
    }
}
void Get_height(){
    int i,j,k=0;
    for(i=0;i<n;i++){
        if(!rk[i])
            height[rk[i]]=0;
        else{
            if(k)
                k--;
            for(j=sa[rk[i]-1];s[i+k]==s[j+k];k++);
            height[rk[i]]=k;
        }
    }
}
void read(){
    scanf("%s",s);
    n=strlen(s);
    int i;
    s[n]='$';
    for(i=1;i<=n;i++)
        s[n+i]=s[n-i];
    hn=n,n=n*2+1;
    for(Log=1;(1<<Log)<=n;Log++);
    Log--;
    s[n]=0;
}
void prepare_st(){
    int i,j;
    for(i=0;i<n;i++)
        *st[i]=height[i];
    for(j=1;j<=Log;j++)
        for(i=0;i<n;i++)
            if(i+(1<<j)<=n)
                st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}
int Get_st(int i,int j){
    if(i>j)
        swap(i,j);
    int t=lg2[j-i];
    return min(st[i+1][t],st[j-(1<<t)+1][t]);
}
void solve(){
    int i,t,t1,j;
    for(i=1;i<=hn;i++){
        for(j=0;j+i<hn;j+=i){
            t=min(Get_st(rk[j+1],rk[j+i+1]),i-1);
            t+=t1=min(Get_st(rk[n-j-i-1],rk[n-j-1]),i);
            if(t>=i){
                cnt[j-t1+1]++;
                cnt[j-t1+1+t-i+1]--;
            }
        }
    }
    for(i=1;i<hn;i++)
        cnt[i]+=cnt[i-1];
    cnt[hn]=0;
    for(i=1;i<=hn;i++)
        cnt[i]+=cnt[i-1];
    for(i=1;i<=hn;i++){
        for(j=0;j+i<hn;j+=i){
            t=min(Get_st(rk[n-j-i-1],rk[n-j-1]),i);
            t+=t1=min(Get_st(rk[j+1],rk[j+i+1]),i-1);
            if(t>=i)
                ans+=cnt[j+i+t1+1]-cnt[j+i+t1+1-(t-i+1)];
        }
    }
}
void prepare(){
    for(int i=2;i<=(MAXN<<1);i++)
        lg2[i]=lg2[i>>1]+1;
}
int main()
{
    prepare();
    Read(T);
    while(T--){
        memset(cnt,0,sizeof cnt);
        ans=0;
        read();
        Get_sa();
        Get_height();
        prepare_st();
        solve();
        printf("%lld\n",ans);
    }
}

outer_form

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
【后缀数组】[NOI2016]优秀的拆分

题目描述如果一个字符串可以被拆分为 AABBAABB 的形式，其中 AA 和 BB 是任意非空字符串，则我们称该字符串的这种拆分是优秀的。例如，对于字符串 aabaabaa，如果令 A=aabA = \mathrm{aab}，B=aB = \mathrm{a}，我们就找到了这个字符串拆分成 AABBAABB 的一种方式。一个字符串可能没有优秀的拆分，也可能存在不止一种优秀的拆分。比如我们令 A=aA
复制链接

扫一扫