bzoj 5073: [Lydsy1710月赛]小A的咒语后缀数组+dp

最新推荐文章于 2019-03-11 07:26:48 发布

原创最新推荐文章于 2019-03-11 07:26:48 发布 · 242 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

后缀数组

本文介绍了一种优化的字符串匹配算法，通过使用后缀数组和LCP（最长公共前缀）来实现快速匹配。该算法能在O(1)的时间复杂度内完成转移，适用于大规模字符串的匹配任务。

题目大意：
给出一个长度为 $n$ 的字符串 $A$ 和一个长度为 $m$ 的字符串 $B$ 。询问是否能从 $A$ 中取出不多于 $k$ 段，使得这些段按原来顺序拼接可以变成字符串 $B$ 。多组数据。
$m≤n≤10^5$ ， $k \leq 100$ ， $T \leq 10$

分析：
我们设 $f [i] [j]$ 表示用 $A$ 的前 $i$ 个字符，取出 $j$ 段，可以拼出的 $B$ 的最长前缀长度。
考虑顺推，两种转移，
$f [i + 1] [j] = f [i] [j]$
$f [i + d] [j + 1] = f [i] [j] + d$
其中， $d = l c p (A . s u f (i + 1), B . s u f (f [i] [j] + 1))$ 。
直接后缀数组就可以 $O (1)$ 转移了。

代码：

/**************************************************************
    Problem: 5073
    User: ypxrain
    Language: C++
    Result: Accepted
    Time:22000 ms
    Memory:100716 kb
****************************************************************/
 
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
 
const int maxn=2e5+7;
 
using namespace std;
 
int n,m,k,T,len;
int x[maxn],y[maxn],c[maxn],f[maxn][101],lg[maxn];
 
char s[maxn];
 
struct suffle_array{
    int h[maxn][20],rank[maxn],sa[maxn];
    void getsa()
    {
        memset(sa,0,sizeof(sa));
        memset(x,0,sizeof(x));
        memset(y,0,sizeof(y));
        int n=len,m=1000;
        for (int i=1;i<=m;i++) c[i]=0;
        for (int i=1;i<=n;i++) x[i]=s[i];
        for (int i=1;i<=n;i++) c[x[i]]++;
        for (int i=1;i<=m;i++) c[i]+=c[i-1];
        for (int i=n;i>0;i--) sa[c[x[i]]--]=i;
        for (int k=1;k<=n;k<<=1)
        {
            int num=0;
            for (int i=n-k+1;i<=n;i++) y[++num]=i;
            for (int i=1;i<=n;i++) if (sa[i]>k) y[++num]=sa[i]-k;
            for (int i=1;i<=m;i++) c[i]=0;
            for (int i=1;i<=n;i++) c[x[i]]++;
            for (int i=1;i<=m;i++) c[i]+=c[i-1];
            for (int i=n;i>0;i--) sa[c[x[y[i]]]--]=y[i],y[i]=0;
            swap(x,y);
            num=1;
            x[sa[1]]=1;
            for (int i=2;i<=n;i++)
            {
                if ((y[sa[i]]!=y[sa[i-1]]) || (y[sa[i]+k]!=y[sa[i-1]+k]))
                {
                    x[sa[i]]=++num;
                }
                else x[sa[i]]=num;
            }
            if (num>=n) break;
            m=num;
        }
        for (int i=1;i<=n;i++) rank[i]=x[i];
    }
    void getheight()
    {
        int n=len,k=0;
        for (int i=1;i<=n;i++)
        {
            if (k) k--;
            int j=sa[rank[i]-1];
            while ((i+k<=n) && (j+k<=n) && (s[i+k]==s[j+k])) k++;
            h[rank[i]][0]=k;
        }
        int c=1;
        for (int j=1;j<20;j++)
        {
            for (int i=1;i+2*c-1<=n;i++) h[i][j]=min(h[i][j-1],h[i+c][j-1]);
            c<<=1;
        }
    }
    int lcp(int x,int y)
    {
        x=rank[x],y=rank[y];
        if (x>y) swap(x,y);
        x++;
        int k=lg[y-x+1];
        return min(h[x][k],h[y-(1<<k)+1][k]);
    }
}A;
 
int main()
{
    for (int i=1;i<=2e5;i++) lg[i]=trunc(log(i+0.5)/log(2));
    scanf("%d",&T);
    while (T--)
    {
        scanf("%d%d%d",&n,&m,&k);
        scanf("%s",s+1);
        s[n+1]=' ';     
        scanf("%s",s+n+2);
        len=n+m+1;      
        A.getsa();
        A.getheight();
        memset(f,0,sizeof(f));
        for (int j=0;j<k;j++)
        {
            for (int i=0;i<n;i++)
            {
                if (f[i][j]==m) break;
                f[i+1][j]=max(f[i+1][j],f[i][j]);
                int d=A.lcp(i+1,f[i][j]+1+n+1);
                f[i+d][j+1]=max(f[i+d][j+1],f[i][j]+d);
            }
        }
        int ans=0;
        for (int j=1;j<=k;j++)
        {
            for (int i=1;i<=n;i++) ans=max(ans,f[i][j]);
        }
        if (ans==m) printf("YES\n");
               else printf("NO\n");
    }
}