CF1537E - Erase and Extend——结论题

最新推荐文章于 2023-10-18 11:05:23 发布

偶耶XJX

最新推荐文章于 2023-10-18 11:05:23 发布

阅读量278

点赞数

分类专栏：信息竞赛解题文章标签： Codeforces 字符串

本文链接：https://blog.csdn.net/weixin_43960287/article/details/118091091

版权

信息竞赛解题专栏收录该内容

194 篇文章 6 订阅

订阅专栏

1537E - Erase and Extend

题目描述

你可以对一个长度为 $n$ 的字符串 $s$ 进行任意次如下操作

去掉 $s$ 末尾的字符。
把 $s$ 赋值为 $s + s$ 。

求一个操作后的、长度恰好为 $k$ 的、字典序最小的字符串。

数据范围与提示

$1\le n,k\le 5\cdot 10^5$ 。

前言

好久没碰到字符串的题了，做起来真舒服。

思路

注意到操作后的字符串一定是若干个 $s$ 的前缀拼起来然后重复，再掐尾的结果。

首先有一个 $\rm Easy\; Version$ 的结论：最终的字符串一定是 $s$ 的某一个前缀经过重复然后掐尾得来的。

考虑用反证法证明：如果是两个长度不相等的前缀拼一起然后重复（假设是 $A B$ 和 $A$ ），那么为什么不选 $(A B) (A B) . . .$ 而选 $(A B A) (A B A) . . .$ 呢？一定是串 $A B A A$ 比串 $A B A B$ 的字典序更小。这说明 $A$ 比 $B$ 的字典序小，那么显然有 $(A) (A) . . .$ 比 $(A B A) (A B A) . . .$ 更优，这就矛盾了。

同理，其它情况也可以用类似方法否定掉，最后只可能是一个前缀的重复最优。

到这里已经可以用暴力过 $Easy\;Version$ 了，而 $Hard\;Version$ 有很多做法，如二分Hash比较、SA等。这里提供一个强结论支持的简单做法。

（为了方便，我们把前缀 $0\sim i$ 经重复后的串叫做前缀构造串 $i$ ）枚举每个前缀，假设 $i$ 以前字典序最小的前缀构造串为 $a$ ，那么把 $s_i$ 和 $s_{i\%a}$ 进行比较：

若 $s_i>s_{i\%a}$ ，直接break；
若 $s_i=s_{i\%a}$ ，不管它；
若 $s_i<s_{i\%a}$ ，把 $a$ 赋值为 $i$ 。

这样最后得到的 $a$ 即为最优的前缀。

证明一下为什么是正确的：

若 $s_i>s_{i\%a}$ ，直接退出循环。此时表明前缀 $i$ 的字典序已经大于构造串 $a$ ，那么 $i$ 后面的前缀的字典序一定也大于 $a$ ，所以后面都不用找了。
若 $s_i=s_{i\%a}$ ，不处理。由于有第一条存在，所以前缀构造串 $a$ 中的字符一定都 $\le s_0$ ，此时分情况讨论：
- 若 $s_i<s_0$ ，那么如果构造串 $a$ 的下一个字符 $s_0$ ，则 $i$ 肯定不优；若 $s_0$ ，则 $i$ 也不优（否则一定有一个 $i$ 以前的前缀优于 $a$ ，矛盾）。
- 若 $s_i=s_0$ ，那么显然 $i - 1$ 比 $i$ 更优。
若 $s_i<s_{i\%a}$ ，把 $a$ 赋值为 $i$ 。由于前面见大于就退出循环的操作，所以构造串 $i$ 与 $a$ 的前 $i - 1$ 个字符一定是相同的，显然此时 $i$ 就比 $a$ 更优，需要更新答案。

这里的讨论最多只嵌套了三层，其实省略了一些，原本需要讨论得更深。迫于篇幅，用“显然”“一定”替之。

所以最终复杂度 $O (n)$ ，最多跑 $31 m s$ 。

代码

#include<cstdio>//JZM YYDS!!!
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<string>
#include<map>
#include<ctime>
#define ll long long
#define MAXN 500005
#define uns unsigned
#define MOD 998244353ll
#define INF 1e15
#define lowbit(x) ((x)&(-(x)))
using namespace std;
inline ll read(){
	ll x=0;bool f=1;char s=getchar();
	while((s<'0'||s>'9')&&s>0){if(s=='-')f^=1;s=getchar();}
	while(s>='0'&&s<='9')x=(x<<1)+(x<<3)+s-'0',s=getchar();
	return f?x:-x;
}
int n,k;
char s[MAXN];
signed main()
{
	n=read(),k=read();
	scanf("%s",s+1);
	int a=1;
	for(int i=2;i<=min(n,k);i++){
		int p=(i-1)%a+1;
		if(s[p]>s[i])a=i;
		else if(s[p]<s[i])break;
	}
	for(int i=1;i<=k;i++)putchar(s[(i-1)%a+1]);
	putchar('\n');
	return 0;
}