bzoj5073 [Lydsy1710月赛]小A的咒语（SA+lcp+dp+贪心）

考虑dp,f[i][j]表示A串前i字符分成j段能匹配的最大B串前缀长度。
那么如果i+1不选，转移到f[i+1][j]

如果i+1选，则我们可以贪心的尽量多匹配一些，即匹配t=lcp(i+1,f[i][j]+1)个,转移到f[i+t][j+1]

我们SA+RMQ预处理一下，O(1)查询lcp即可。

复杂度$O(T(nk+nlogn))$
注意多组数据的一些细节，比如rk要清零囧

第1000题！纪念一下嘿嘿嘿qaq

算是完成了自己定下的省选前的目标吧！后天大后天希望能够一切顺利！

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 200010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int a[N],rk[N<<1],rk1[N<<1],sa[N],h[N],tmp[N],cnt[N],Log[N],st[N][20];
int f[N>>1][110];
char s[N];
inline void SA(){
    int n=0,m=26;memset(rk,0,sizeof(rk));memset(rk1,0,sizeof(rk1));
    scanf("%s",s+1);int len=strlen(s+1);
    for(int i=1;i<=len;++i) a[++n]=s[i]-'a'+1;
    scanf("%s",s+1);len=strlen(s+1);a[++n]=++m;
    for(int i=1;i<=len;++i) a[++n]=s[i]-'a'+1;
    for(int i=0;i<=m;++i) cnt[i]=0;
    for(int i=1;i<=n;++i) cnt[a[i]]=1;
    for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
    for(int i=1;i<=n;++i) rk[i]=cnt[a[i]];
    int k=0;
    for(int p=1;k!=n;m=k,p+=p){
        for(int i=0;i<=m;++i) cnt[i]=0;
        for(int i=1;i<=n;++i) cnt[rk[i+p]]++;
        for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
        for(int i=n;i>=1;--i) tmp[cnt[rk[i+p]]--]=i;
        for(int i=0;i<=m;++i) cnt[i]=0;
        for(int i=1;i<=n;++i) cnt[rk[tmp[i]]]++;
        for(int i=1;i<=m;++i) cnt[i]+=cnt[i-1];
        for(int i=n;i>=1;--i) sa[cnt[rk[tmp[i]]]--]=tmp[i];
        memcpy(rk1,rk,sizeof(rk)>>1);rk[sa[1]]=k=1;
        for(int i=2;i<=n;++i){
            if(rk1[sa[i]]!=rk1[sa[i-1]]||rk1[sa[i]+p]!=rk1[sa[i-1]+p]) ++k;
            rk[sa[i]]=k;
        }
    }Log[0]=-1;
    for(int i=1;i<=n;++i){
        if(rk[i]==1){h[1]=0;continue;}
        if(i==1||h[rk[i-1]]<=1) k=0;
        if(k) --k;
        while(i+k<=n&&a[i+k]==a[sa[rk[i]-1]+k]) ++k;
        h[rk[i]]=k;
    }for(int i=1;i<=n;++i) Log[i]=Log[i>>1]+1,st[i][0]=h[i];
    for(int i=1;i<=Log[n];++i)
        for(int j=1;j<=n;++j){
            if(j+(1<<i-1)>n) break;
            st[j][i]=min(st[j][i-1],st[j+(1<<i-1)][i-1]);
        }
}
inline int lcp(int x,int y){
    x=rk[x];y=rk[y];if(!y) return 0;if(x>y) swap(x,y);x++;
    int d=Log[y-x+1];return min(st[x][d],st[y-(1<<d)+1][d]);
}
int main(){
//  freopen("a.in","r",stdin);
    int tst=read();
    while(tst--){
        int n=read(),m=read(),K=read();SA();
        memset(f,0,sizeof(f));
        for(int i=0;i<n;++i)
            for(int j=0;j<=K;++j){
                f[i+1][j]=max(f[i+1][j],f[i][j]);
                int t=lcp(i+1,f[i][j]+n+1+1);
                f[i+t][j+1]=max(f[i+t][j+1],f[i][j]+t);
            }int ans=0;
        for(int i=0;i<=K;++i) ans=max(ans,f[n][i]);
        puts(ans==m?"YES":"NO");
    }return 0;
}