2017 icpc 沈阳 G

题目大意：有n个点， 每个点有一个数字0 - 9， 第 i 个点只能到 第(i * i + 1)个点，问你在哪个点出发走n次构成的数字串最大。

思路：利用求后缀数组的倍增比较思想， 许多细节需要注意。

#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mk make_pair
#define pii pair<int,int>
#define piii pair<int, pair<int,int>>
using namespace std;

const int N=2e5+7;
const int M=1e4+7;
const int inf=0x3f3f3f3f;
const LL INF=0x3f3f3f3f3f3f3f3f;
const int mod=1e9 + 7;

int n, tot, a[N], nx[N][19], sa[N], t[N], t2[N], c[N], nxk[N], ans[N];
vector<int> prek[N];
void buildSa(int n, int m) {
    int i, j, *x = t, *y = t2;
    for(i = 0; i < m; i++) c[i] = 0;
    for(i = 0; i < n; i++) c[x[i] = a[i]]++, nxk[i] = i;
    for(i = 1; i < m; i++) c[i] += c[i - 1];
    for(i = n - 1; i >= 0; i--) sa[--c[x[i]]] = i;

    for(int k = 1; k <= n; k <<= 1) {
        for(i = 0; i < n; i++) nxk[i] = nx[nxk[i]][0], prek[i].clear();
        for(i = 0; i < n; i++) prek[nxk[i]].push_back(i);
        int p = 0;
        for(i = 0; i < n; i++) {
            for(j = 0; j < prek[sa[i]].size(); j++)
                y[p++] = prek[sa[i]][j];
        }
        for(i = 0; i < m; i++) c[i] = 0;
        for(i = 0; i < n; i++) c[x[y[i]]]++;
        for(i = 1; i < m; i++) c[i] += c[i - 1];
        for(i = n - 1; i >= 0; i--) sa[--c[x[y[i]]]] = y[i];
        swap(x, y);
        p = 1; x[sa[0]] = 0;
        for(i = 1; i < n; i++) {
            if(y[sa[i - 1]] == y[sa[i]] && y[nxk[sa[i - 1]]] == y[nxk[sa[i]]])
                    x[sa[i]] = p - 1;
            else x[sa[i]] = p++;
        }
        if(p >= n) break;
        m = p;
    }
}
int main() {
    int T; scanf("%d", &T);
    for(int cas = 1; cas <= T; cas++) {
        scanf("%d", &n);
        for(int i = 0; i < n; i++) {
            scanf("%1d", &a[i]);
        }

        for(int i = 0; i < n; i++) {
            nx[i][0] =  (1ll * i * i + 1) % n;
        }

        for(int j = 1; j < 19; j++) {
            for(int i = 0; i < n; i++) {
                nx[i][j] = nx[nx[i][j - 1]][j - 1];
            }
        }
        buildSa(n, 10);
        tot = 0;
        int now = sa[n - 1];
        for(int i = 1; i <= n; i++) {
            ans[tot++] = a[now];
            now = nx[now][0];
        }
        printf("Case #%d: ", cas);
        for(int i = 0; i < tot; i++)
            printf("%d", ans[i]);
        puts("");
    }
    return 0;
}
/*
*/