20200129模拟赛T1 string

题目描述：

分析：

看到题目：后缀数组，二分，加加减减
然后。。。然后。。。

“后缀数组是个好东西，我有头发的时候天天写。”

我们首先要求出二分的区间，可能二分到所有子串字典序编号
二分的过程当中，首先先找出字典序当前值得字符串，这里要用到height数组RMQ
我们现在就需要将所有字典序大于该子串的切掉，从前往后枚举开头i，当LCP大于目标串的Len时，说明字典序更大，这时候要将i+Len位置切掉
最后通过切的次数来判断二分

以上说得很好。。。
真就写20分钟，调两个小时

“后缀数组是个好东西，我有头发的时候天天写。”

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>

#define maxn 200005

using namespace std;

inline int getint()
{
	int num=0,flag=1;char c;
	while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
	while(c>='0'&&c<='9')num=num*10+c-48,c=getchar();
	return num*flag;
}

int n,k;
char s[maxn];
int st,len;
int sa[maxn],height[maxn];
int rk[maxn],bit[maxn],f[21][maxn];
int cnt[maxn],x[maxn],y[maxn];
int aa[maxn];

inline void suffix()
{
	memset(cnt,0,sizeof cnt);
	for(int i=1;i<=n;i++)cnt[int(s[i])]++;
	for(int i=1;i<=128;i++)cnt[i]+=cnt[i-1];
	for(int i=n;i>=1;i--)sa[cnt[int(s[i])]--]=i;
	rk[sa[1]]=1;
	for(int i=2;i<=n;i++)rk[sa[i]]=rk[sa[i-1]]+(s[sa[i]]!=s[sa[i-1]]);
	for(int k=1;rk[sa[n]]!=n;k<<=1)
	{
		for(int i=1;i<=n;i++) 
		{
			x[i]=rk[i];
			y[i]=(i+k<=n)?rk[i+k]:0;
		}
		memset(cnt,0,sizeof cnt);
		for(int i=1;i<=n;i++)cnt[y[i]]++;
		for(int i=1;i<=n;i++)cnt[i]+=cnt[i-1];
		for(int i=n;i>=1;i--)rk[cnt[y[i]]--]=i;
		memset(cnt,0,sizeof cnt);
		for(int i=1;i<=n;i++)cnt[x[i]]++;
		for(int i=1;i<=n;i++)cnt[i]+=cnt[i-1];
		for(int i=n;i>=1;i--)sa[cnt[x[rk[i]]]--]=rk[i];
		rk[sa[1]]=1;
		for(int i=2;i<=n;i++)rk[sa[i]]=rk[sa[i-1]]+(x[sa[i]]!=x[sa[i-1]]||y[sa[i]]!=y[sa[i-1]]);
	}
	int now=0;
	for(int i=1;i<=n;i++)
	{
		if(now)now--;
		for(int j=sa[rk[i]+1];s[j+now]==s[i+now];now++);
		height[rk[i]]=now;
	}
	for(int i=1;i<=n;i++)
	{
		f[0][i]=height[i];
		bit[i]=bit[i-1];
		if(i>=(1<<(bit[i]+1)))bit[i]++;
	}
	for(int p=1;p<20;p++)
		for(int i=1,j=(1<<(p-1))+1;j<=n;i++,j++)
			f[p][i]=min(f[p-1][i],f[p-1][j]);
	for(int i=1;i<=n;i++)aa[i]=rk[i];
}

inline int LCP(int x,int y)
{
	if(x==y)return n-x+1;
	x=rk[x],y=rk[y];
	if(x>y)swap(x,y);
	int tmp=bit[y-x];
	return min(f[tmp][x],f[tmp][y-(1<<tmp)]);
}

inline long long getl()
{
	long long ans=1;
	for(int i=1;s[sa[i]]!=s[sa[n]];i++)
		ans+=n-sa[i]+1-height[i];
	return ans;
}

inline long long getr()
{
	long long ans=0;
	for(int i=1;i<=n;i++)
		ans+=n-sa[i]+1-height[i];
	return ans;
}

inline void solve(long long num)
{
	int pos;
	for(pos=1;n-sa[pos]+1<num;pos++)
		num-=n-sa[pos]+1-height[pos];
	st=sa[pos];len=num;
}

inline bool check(int k)
{
	if(len==1)return 0;
	int ans=0,ed=n;
	for(int i=1;i<=n;i++)
	{
		int tmp=LCP(i,st);
		if(tmp>=len)ed=min(ed,i+len-2);
		else if(s[i+tmp]>s[st+tmp])ed=min(ed,i+tmp-1);
		if(i==ed)ans++,ed=n;
	}
	return ans<=k;
}


int main()
{
	k=getint();
	scanf("%s",s+1);
	n=strlen(s+1);
	suffix();
	for(int i=1;i<=n;i++)rk[i]=aa[i];
	long long l=getl(),r=getr();
	while(l<r)
	{
		long long mid=(l+r+1)/2;
		solve(mid);
		if(check(k))r=mid-1;
		else l=mid;
	}
	solve(l);
	s[st+len]=0;
	printf("%s
",s+st);
}