题目大意:
给出一个长度为
n
n
n的字符串
A
A
A和一个长度为
m
m
m的字符串
B
B
B。询问是否能从
A
A
A中取出不多于
k
k
k段,使得这些段按原来顺序拼接可以变成字符串
B
B
B。多组数据。
m
≤
n
≤
1
0
5
m≤n≤10^5
m≤n≤105,
k
≤
100
k≤100
k≤100,
T
≤
10
T≤10
T≤10
分析:
我们设
f
[
i
]
[
j
]
f[i][j]
f[i][j]表示用
A
A
A的前
i
i
i个字符,取出
j
j
j段,可以拼出的
B
B
B的最长前缀长度。
考虑顺推,两种转移,
f
[
i
+
1
]
[
j
]
=
f
[
i
]
[
j
]
f[i+1][j]=f[i][j]
f[i+1][j]=f[i][j]
f
[
i
+
d
]
[
j
+
1
]
=
f
[
i
]
[
j
]
+
d
f[i+d][j+1]=f[i][j]+d
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
)
)
d=lcp(A.suf(i+1),B.suf(f[i][j]+1))
d=lcp(A.suf(i+1),B.suf(f[i][j]+1))。
直接后缀数组就可以
O
(
1
)
O(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");
}
}