BZOJ2806:[CTSC2012]Cheat(广义SAM,二分,DP)

Description

Input

第一行两个整数N，M表示待检查的作文数量，和小强的标准作文库的行数
接下来M行的01串，表示标准作文库
接下来N行的01串，表示N篇作文

Output

N行，每行一个整数，表示这篇作文的L0值。

Sample Input

1 2
10110
000001110
1011001100

Sample Output

4

HINT

输入文件不超过1100000字节

注意：题目有改动,可识别的长度不小于90%即可，而不是大于90%

Solution

首先把广义$SAM$建出来，然后考虑对于一个询问，

我们可以把这个作文放到$SAM$上跑，就可以求得这个字符串的每个位置向前延伸仍然可以匹配的最大长度，记为$Len[i]$。

因为直接求答案不好求，所以我们二分一个答案$lim$

剩下的就是$DP$了。$f[i]$表示到第$i$个位置最多能匹配多少，转移显然。

$f[i]=max(f[i-1]),\sum_{j=lim}^{len} f[i-j]+j$。 单调队列优化一下即可。

Code

#include<iostream>
#include<cstring>
#include<cstdio>
#define N (2200009)
using namespace std;

int n,m,len,Len[N],f[N],q[N];
char s[N];

struct SAM
{
    int son[N][2],fa[N],step[N];
    int p,q,np,nq,last,cnt;
    SAM(){last=cnt=1;}

    void Insert(int x)
    {
        p=last; np=last=++cnt; step[np]=step[p]+1;
        while (p && !son[p][x]) son[p][x]=np,p=fa[p];
        if (!p) fa[np]=1;
        else
        {
            q=son[p][x];
            if (step[q]==step[p]+1) fa[np]=q;
            else
            {
                nq=++cnt; step[nq]=step[p]+1;
                memcpy(son[nq],son[q],sizeof(son[q]));
                fa[nq]=fa[q]; fa[q]=fa[np]=nq;
                while (son[p][x]==q) son[p][x]=nq,p=fa[p];
            }
        }
    }
    void Find(char s[])
    {
        int now=1,len=0;
        for (int i=0,l=strlen(s); i<l; ++i)
        {
            int c=s[i]-'0';
            if (son[now][c]) ++len,now=son[now][c];
            else
            {
                while (now && !son[now][c]) now=fa[now];
                if (!now) len=0,now=1;
                else len=step[now]+1,now=son[now][c];
            }
            Len[i+1]=len;
        }
    }
}SAM;

bool check(int lim)
{
    int head=1,tail=0,len=strlen(s);
    for (int i=1; i<=len; ++i)
    {
        f[i]=f[i-1];
        if (i-lim<0) continue;
        while (head<=tail && f[q[tail]]-q[tail]<=f[i-lim]-i+lim) --tail;
        q[++tail]=i-lim;
        while (head<=tail && q[head]<i-Len[i]) ++head;
        if (head<=tail) f[i]=max(f[i],f[q[head]]+i-q[head]);
    }
    return f[len]*10>=len*9;
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1; i<=m; ++i,SAM.last=1)
    {
        scanf("%s",s);
        for (int j=0,l=strlen(s); j<l; ++j)
            SAM.Insert(s[j]-'0');
    }
    for (int i=1; i<=n; ++i)
    {
        scanf("%s",s);
        SAM.Find(s);
        int l=1,r=strlen(s),ans=0;
        while (l<=r)
        {
            int mid=(l+r)>>1;
            if (check(mid)) ans=mid,l=mid+1;
            else r=mid-1;
        }
        printf("%d\n",ans);
    }
}