百度之星 数字串

题目描述:给出一个不包含字符0长度为n的字符串,请往该字符串中添加恰好k个'+'变成表达式,并使这个表达式的结果尽可能的大。n给到1e6。

k个加号就是整体分成k+1段

很明显有结论:选出一个长为n-k的段,剩下的部分一个数字一段

选出一个长为n-k的段要尽可能大,用后缀数组就行了

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int n,m,k,pos;
const int N=1000010;
char s[N];;
int x[N],y[N],c[N],sa[N],ans[N];
void Write(int x,int opt)
{
	if(opt&&!x)putchar('0');
	if(!x)return;
	Write(x/10,0);
	putchar((x-x/10*10)+'0');
}
void SA()
{
	for(int i=1;i<=n;++i)++c[x[i]=s[i]];
	for(int i=2;i<=m;++i)c[i]+=c[i-1];
	for(int i=n;i>=1;--i)sa[c[x[i]]--]=i;
	for(int k=1;k<=n;k<<=1)
	{
		int cnt=0;
		for(int i=n-k+1;i<=n;++i)y[++cnt]=i;
		for(int i=1;i<=n;++i)
			if(sa[i]>k)y[++cnt]=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=2;i<=m;++i)c[i]+=c[i-1];
		for(int i=n;i>=1;--i)sa[c[x[y[i]]]--]=y[i],y[i]=0;
		swap(x,y);
		x[sa[1]]=1;cnt=1;
		for(int i=2;i<=n;++i)x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?cnt:++cnt;
		if(cnt==n)break;
		m=cnt;
	}
}
int main()
{
	cin>>n>>k;
	scanf("%s",s+1);
	m=122;
	SA();
	for(int i=n;i>=1;--i)
		if(sa[i]<=k+1)
		{
			pos=sa[i];
			break;
		}
	for(int i=pos;i<=pos+n-k-1;++i)ans[i-pos+1]=s[i]-'0';
	for(int i=1;i<=pos-1;++i)ans[n-k]+=s[i]-'0';
	for(int i=pos+n-k;i<=n;++i)ans[n-k]+=s[i]-'0';
	for(int i=n-k;i>=1;--i)ans[i-1]+=ans[i]/10,ans[i]%=10;
	if(ans[0])cout<<ans[0];
	for(int i=1;i<=n-k;++i)printf("%d",ans[i]);
	return 0;
}
/*
20 19
99999999999999999999
*/
posted @ 2023-11-09 15:27  wljss  阅读(74)  评论(0编辑  收藏  举报