404 最小循环覆盖2

// 404 最小循环覆盖2.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

/*
http://oj.daimayuan.top/course/22/problem/935

给你一个字符串 a
，你需要求出这个字符串的字典序最小的最小循环覆盖。

b是 a 的最小循环覆盖，当且仅当 a是通过 b复制多次并连接后得到的字符串的子串，
且 b 是满足条件的字符串中长度最小的。你需要找出字典序最小的最小循环覆盖。

输入一个字符串 a，输出一个字符串表示字典序最小的最小循环覆盖。

输入格式
一行一个字符串 a。

输出格式
输出一行一个字符串表示答案。

样例输入1
bcabcabcabcab
样例输出1
abc
样例输入2
aaaaaaa
样例输出2
a
数据规模
对于所有数据，保证 1≤|a|≤105
，字符串均由小写字母构成。
*/
#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;


const int N = 200010;
char s[N];
char cs[N];
int ne[N];
int n, m;

string getmin(char s[]) {
	int n = strlen(s + 1);
	for (int l = 1; l <= n; l++)
		s[l + n] = s[l];
	int i = 1, j = 2;
	while (j <= n) {
		int k = 0;
		while (k < n && s[i + k] == s[j + k])
			++k;
		if (s[i + k] > s[j + k])
			i += k + 1;
		else
			j += k + 1;
		if (i == j)
			++j;
		if (i > j)
			swap(i, j);
	}
	string res = "";
	for (int l = i; l <= i + n -1; l++)
		res += s[l];
	res = res.substr(0, res.size() / 2);
	return res;
}

//kmp求出最小循环节  然后最小表示法求出字母序最小的写法
int main()
{
	cin >> s + 1;
	m = strlen(s + 1);
	ne[1] = 0;
	for (int i = 2, j = 0; i <= m; i++) {
		while (j && s[i] != s[j + 1]) j = ne[j];
		if (s[i] == s[j + 1]) j++;
		ne[i] = j;
	}

	m = m - ne[m];
	for (int i = 1; i <= m; i++) {
		cs[i] = s[i]; cs[i + m] = s[i];
	}
	cout << getmin(cs);


    return 0;
}