题意:将一个字符串切成k块,使得字典序最大的那块最小。
ORZ WJMZBMR,几行题解读了一天才懂。
快速比较两个子串的大小可以利用LCP(最长公共前缀),比较公共前缀的下一个字符的大小就够了。
利用这种思想,首先我们可以预处理所有子串的LCP(后缀数组+记录 O(2nlog(2n))+O(n*n),dp(O(4*n*n)))
然后将这些子串利用LCP按照字典序排序,开始二分答案。
二分的答案就是这K个块字典序的上限。假设以i作为起点,由于字典序上限已知,所以我们可以立刻求出i点最远能选到哪个点。
现在问题变成了:已知每一个点最远能跳R的距离,求是否存在一条路径,使得跳K次回到起点。
首先我们假设,每个点的R≠0,意思就是每个点都能向后跳,这样我们只需要用贪心的思想,枚举任意点为起点,然后向后能跳多少跳多少,若跳回来所花的次数T<=K,则为true 【由于每个点都能向后跳,则我们一定能够通过改变几个跳跃,使得T==K 若当前的点数<K,也就是每次只跳1步都没法跳到,当然就是false了】
现在的问题就是,如果有某些点R==0怎么办,也就是它一步也不能向后跳,并且其它位置也不能跳到这个位置。所以我们想到,将这个点删除掉,并且,将所有受到影响的点全部减一,以前可以从 a向后跳3步,但现在 b (a<=b<=a+3)被删掉了,所以a只能挑2步了,这样最多迭代n次 处理之后,所有点的R都是不为0的了。
代码写的很挫。。。
#include <cstdio>
#include <iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
#define maxn 2005
char str[maxn];
int sa[maxn],t1[maxn],t2[maxn],c[maxn],n;
void suffix(int m)
{
int *x=t1,*y=t2;
for(int i=0; i<m; i++)c[i]=0;
for(int i=0; i<n; i++)c[x[i]=str[i]]++;
for(int i=1; i<m; i++)c[i]+=c[i-1];
for(int i=n-1; i>=0; i--)sa[--c[x[i]]]=i;
for(int k=1; k<=n; k<<=1)
{
int p=0;
for(int i=n-k; i<n; i++)y[p++]=i;
for(int i=0; i<n; i++)if(sa[i]>=k)y[p++]=sa[i]-k;
for(int i=0; i<m; i++)c[i]=0;
for(int i=0; i<n; i++)c[x[y[i]]]++;
for(int i=0; i<m; i++)c[i]+=c[i-1];
for(int i=n-1; i>=0; i--)sa[--c[x[y[i]]]]=y[i];
swap(x,y);
p=1;
x[sa[0]]=0;
for(int i=1; i<n; i++)
x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
if(p>=n)break;
m=p;
}
}
int rank[maxn],height[maxn];
void getheight()
{
int k=0;
for(int i=0; i<n; i++)rank[sa[i]]=i;
for(int i=0; i<n; i++)
{
if(k)k--;
if(!rank[i])continue;
int j=sa[rank[i]-1];
while(str[i+k]==str[j+k])k++;
height[rank[i]]=k;
}
}
int f[maxn][30];
void RMQINIT()
{
for(int i=0;i<n;i++)
f[i][0]=height[i];
for(int j=1;(1<<j)<=n;j++)
for(int i=0;i+(1<<j)-1<n;i++)
f[i][j]=min(f[i][j-1],f[i+(1<<(j-1))][j-1]);
}
int RMQ(int l,int r)
{
if(l>r)swap(l,r);
l++;
int k=floor(log(r-l+1.0)/log(2.0));
return min(f[l][k],f[r+1-(1<<k)][k]);
}
int tn,k,st,mid;
int lcp(int l,int r)
{
if(l==r) return n-l;
return RMQ(rank[l],rank[r]);
}
int Lcp[1015][1015];
struct node
{
int l,r;
int size;
bool operator <(const node &x) const
{
int LCP=min(size,Lcp[l][x.l]);
LCP=min(LCP,x.size);
char a = LCP<size?str[l+LCP]:0;
char b = LCP<x.size?str[x.l+LCP]:0;
return a<b;
}
}a[1111111];
vector<int> v[maxn];
void debug(int pos)
{
for(int i=a[pos].l;i<a[pos].r;i++)
{
putchar(str[i]);
}
puts("");
}
bool vis[maxn];
vector<int>far;
bool cal()
{
far.clear();
for(int i=0;i<tn;i++)
{
if(i==a[mid].l)
{
far.push_back(a[mid].size);
continue;
}
int LCP=min(tn,Lcp[i][a[mid].l]);
LCP=min(LCP,a[mid].size);
if(LCP>=tn)
{
far.push_back(tn);
continue;
}
else
{
if(str[i+LCP]<str[a[mid].l+LCP]) far.push_back(tn);
else far.push_back(LCP);
}
}
int ok=1;
while(ok)
{
ok=0;
for(int i=0;i<far.size();i++)
{
if(far[i]==0)
{
for(int j=0;j<far.size();j++)
{
if(j<i&&i<=j+far[j]) far[j]--;
else if(j>i&&j+far[j]>=i+far.size()) far[j]--;
}
ok=1;
far.erase(i+far.begin());
}
}
}
if(far.size()<k) return false;
int len=far.size();
for(int i=0;i<len;i++) far.push_back(far[i]);
for(int i=0;i<len;i++)
{
int times=0;
for(int j=i;j<i+len;j+=far[j]) times++;
if(times<=k) return true;
}
return false;
}
int main()
{
int cas;
scanf("%d",&cas);
while(cas--)
{
scanf("%d%d",&n,&k);
scanf("%s",str);
tn=n;
if(k==1) {puts(str);continue;}
for(int i=n;i<2*n;i++) str[i]=str[i-n];
n*=2;
str[n]=0;
n++;
suffix(128);
getheight();
RMQINIT();
n--;
int top=0;
for(int i=0;i<tn;i++)
{
for(int j=i;j<i+tn;j++)
{
a[++top].l=i;
a[top].r=j+1;
a[top].size=j+1-i;
}
}
for(int i=0;i<=tn;i++)
for(int j=0;j<=tn;j++)
Lcp[i][j]=lcp(i,j);
sort(a+1,a+top+1);
int l=1,r=top,ans;
while(l<=r)
{
mid=(l+r)/2;
if(cal())
{
ans=mid;
r=mid-1;
}
else
{
l=mid+1;
}
}
debug(ans);
}
return 0;
}