ROI 2019 Day2 花环 题解

题目传送门

题目大意： 给出一个 $01$ 序列，要求删除尽可能少的数字，使得剩下的序列形如 $A1A1A...1A1A$，其中 $A$ 是若干个 $0$，以及要求至少有一个 $1$。

题解

先记录下所有由 $0$ 构成的小段，然后枚举 $A$ 的长度 $i$，考虑贪心，从左到右枚举 $0$ 段，假如当前段长度不足 $i$，就把下一段和当前段接在一起，还不够就继续接，直到够为止，然后再去考虑下一小段。

时间复杂度 $O(n\ln n)$，代码如下：

#include <cstdio>
#define maxn 500010

int n,m=0,a[maxn],pre[maxn],id[maxn];
char s[maxn];

int main()
{
	scanf("%d",&n);scanf("%s",s+1);
	for(int i=1,last0=0,last1=0;i<=n;i++)
	if(s[i]=='0')last0=i,a[++m]=i,pre[i]=last1,id[i]=m;
	else last1=i,pre[i]=last0;
	int ans1=n-m+1,ans2=0;
	for(int i=1;i<=m;i++)
	{
		int now=a[m-i+1],tot=1;
		while(pre[now]&&pre[pre[now]]&&id[pre[pre[now]]]>=i)
		now=a[id[pre[pre[now]]]-i+1],tot++;
		if(tot>1&&tot-1+tot*i>ans1-1+ans1*ans2)ans1=tot,ans2=i;
	}
	printf("%d\n",ans1-1+ans1*ans2);
	for(int i=1;i<=ans2;i++)printf("0");
	for(int i=1;i<ans1;i++){
		printf("1");
		for(int j=1;j<=ans2;j++)printf("0");
	}
}