「JSOI2015」串分割

「JSOI2015」串分割

传送门

首先我们会有一个贪心的想法：分得越均匀越好，因为长的绝对比短的大。

那么对于最均匀的情况，也就是 \(k | n\) 的情况，我们肯定是通过枚举第一次分割的位置，然后每一段长度 \(\frac{n}{k}\) 最后取最小的。

把这个思想运用到一般情况：如果分出来两段长短不一，那么长的只会比短的那个长度多 \(1\) ，再仔细想想，所有段只会有两种不同的长度 \(\lfloor \frac{n}{k} \rfloor, \lceil \frac{n}{k} \rceil\) ，那么我们就只需要判断什么情况下取长的，什么时候取短的就好了。

于是考虑二分最大的那个段的大小，然后再枚举第一次分割的位置，对于当前我们这段，如果分成长段会超过二分值，就换成短的，不然就按长的分，是否符合条件就判一下分出来的段的长度之和的大小，如果大于 \(n\) 说明当前的二分值大了，否则就是小了。

至于一段数的大小，我们可以直接将原数字环变成链然后倍长跑一遍后缀排序，比较的时候就直接拿 \(rank_i\) 来比就是了。

参考代码：

#include <algorithm>

#include <cstdio>

#define rg register

#define file(x) freopen(x".in", "r", stdin), freopen(x".out", "w", stdout)

using namespace std;

template < class T > inline void read(T& s) {

    s = 0; int f = 0; char c = getchar();

    while ('0' > c || c > '9') f |= c == '-', c = getchar();

    while ('0' <= c && c <= '9') s = s * 10 + c - 48, c = getchar();

    s = f ? -s : s;

}

const int _ = 4e5 + 5;

int n, k, a[_], sa[_], rk[_];

inline void sufsort(int n) {

#define cmp(i, j, k) (y[i] == y[j] && y[i + k] == y[j + k])

    static int x[_], y[_], cnt[_];

    int m = 57;

    for (rg int i = 1; i <= n; ++i) ++cnt[x[i] = a[i]];

    for (rg int i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];

    for (rg int i = n; i >= 1; --i) sa[cnt[x[i]]--] = i;

    for (rg int k = 1; k <= n; k <<= 1) {

	    int p = 0;

	    for (rg int i = n - k + 1; i <= n; ++i) y[++p] = i;

	    for (rg int i = 1; i <= n; ++i) if (sa[i] > k) y[++p] = sa[i] - k;

	    for (rg int i = 1; i <= m; ++i) cnt[i] = 0;

	    for (rg int i = 1; i <= n; ++i) ++cnt[x[i]];

	    for (rg int i = 1; i <= m; ++i) cnt[i] += cnt[i - 1];

	    for (rg int i = n; i >= 1; --i) sa[cnt[x[y[i]]]--] = y[i];

	    swap(x, y), x[sa[1]] = p = 1;

	    for (rg int i = 2; i <= n; ++i) x[sa[i]] = cmp(sa[i], sa[i - 1], k) ? p : ++p;

	    if (p == n) break ; else m = p;

    }

    for (rg int i = 1; i <= n; ++i) rk[sa[i]] = i;

}

inline bool check(int mid) {

    int x = n / k + (n % k != 0);

    for (rg int i = 1; i <= x; ++i)

	    for (rg int p = i, j = 1; j <= k; ++j) {

    	    p += rk[p] <= mid ? x : x - 1;

	        if (p >= i + n) return 1;

	    }

    return 0;

}

int main() {

#ifndef ONLINE_JUDGE

    file("cpp");

#endif

    read(n), read(k);

    for (rg int i = 1; i <= n; ++i) scanf("%1d", a + i), a[n + i] = a[i];

    sufsort(n << 1);

    int l = 1, r = n << 1;

    while (l < r) {

	    int mid = (l + r) >> 1;

	    if (check(mid)) r = mid; else l = mid + 1;

    }

    for (rg int i = 1; i <= n; ++i)

	    if (rk[i] == l) {

    	    for (rg int j = 1; j <= n / k + (n % k != 0); ++j) putchar(a[i + j - 1] + '0'); break ;

    	}

    return 0;

}