spoj SUBLEX (Lexicographical Substring Search) RE的欢迎来看看

SPOJ.com - Problem SUBLEX

　　这么裸的一个SAM，放在了死破OJ上面就是个坑。

注意用SAM做的时候输出要用一个数组存下来，然后再puts，不然一个一个字符输出会更慢。

还有一个就是不要多数据输入，估计最后多了几个没用的数字，反正我这么做一直无端端的RE。（就这样浪费了我一天好么！出数据的人这么不负责！）

最后就是，第k大的k是会超过子串数的。（这什么脑残配置？）

　　综上，这题除了坑就是坑。

代码如下：

#include <bits/stdc++.h>
using namespace std;

const int N = 222222;
const int LAST = 90000;
const int K = 26;

struct Node {
    Node *nx[K], *fail;
    int dist;
    long long sub;

    void Clear(const int d = 0) {
        memset(nx, 0, sizeof nx);
        fail = 0;
        dist = d;
        sub = 0;
    }
} ;

struct SAM {
    Node node[N << 1];
    Node *root, *last;
    int ttNode;

    Node *Mem(const int d = 0) {
        Node *temp = node + ttNode++;

        temp->Clear(d);

        return temp;
    }

    void Clear() {
        ttNode = 0;
        root = last = Mem();
    }

    void Expand(const char c) {
        const int idx = c - 'a';
        Node *p = last, *np = Mem(p->dist + 1);

        for ( ; p && p->nx[idx] == 0; p = p->fail) {
            p->nx[idx] = np;
        }
        if (p) {
            Node *q = p->nx[idx];

            if (p->dist + 1 != q->dist) {
                Node *nq = Mem();

                *nq = *q;
                nq->dist = p->dist + 1;
                q->fail = np->fail = nq;
                for ( ; p && p->nx[idx] == q; p = p->fail) {
                    p->nx[idx] = nq;
                }
            } else {
                np->fail = q;
            }
        } else {
            np->fail = root;
        }
        last = np;
    }

    int dist[N << 1];
    Node *ptr[N << 1];

    void GetSub() {
        memset(dist, 0, sizeof dist);
        for (int i = 0; i < ttNode; ++i) {
            ++dist[node[i].dist];
        }
        for (int i = 1; i < ttNode; ++i) {
            dist[i] += dist[i - 1];
        }
        for (int i = 0; i < ttNode; ++i) {
            ptr[--dist[node[i].dist]] = node + i;
        }
        for (int i = ttNode - 1; i >= 0; --i) {
            Node *p = ptr[i];

            p->sub = 1;
            for (int j = 0; j < K; ++j) {
                if (p->nx[j]) {
                    p->sub += p->nx[j]->sub;
                }
            }
        }
        --node[0].sub;
        //for (int i = 0; i < ttNode; ++i) { cout << node[i].dist << ' '; } cout << endl;
        //for (int i = 0; i < ttNode; ++i) { cout << node[i].sub << ' '; } cout << endl;
        //for (int i = 0; i < ttNode; ++i) { cout << i << ": "; for (int j = 0; j < K; ++j) { cout << (node[i].nx[j] ? node[i].nx[j] - node : -1) << ' '; } cout << endl; }
    }
} sam;

char s[N], answer[N];

void Generate(char *const s) {
    srand(time(0));
    for (int i = 0; i < LAST; ++i) {
        s[i] = rand() % 26 + 'a';
    }
    s[LAST] = 0;
}

int Run() {
    //while (~scanf("%s", s)) {
    //while (1) {
        //Generate(s);
        scanf("%s", s);
        sam.Clear();
        for (int i = 0; s[i]; ++i) {
            sam.Expand(s[i]);
        }
        sam.GetSub();
        //cout << sam.root->sub << endl;
        //if (sam.ttNode >= (N << 1)) { puts("???"); while (1) ; }

        int n, k;

        scanf("%d", &n);
        while (n--) {
            Node *p = sam.root;
            int pos = 0;

            scanf("%d", &k);
            //if (k > sam.root->sub) { puts("..."); while (1) ; }
            k = (k - 1) % sam.root->sub + 1;
            while (k > 0) {
                for (int i = 0; i < K; ++i) {
                    if (p->nx[i] == 0) {
                        continue;
                    }

                    const int cnt = p->nx[i]->sub;

                    if (cnt >= k) {
                        //putchar('a' + i);
                        answer[pos++] = 'a' + i;
                        p = p->nx[i];
                        --k;
                        break;
                    } else {
                        k -= cnt;
                    }
                }
            }
            answer[pos] = 0;
            puts(answer);
            //puts("");
        }
    //}

    return 0;
}

int main() {
    //ios::sync_with_stdio(0);
    return Run();
}

UPD：还有更坑的，我开99999 * 2的SAM节点数是会TLE的，开222222 * 2才AC。我猜肯定是新增的数据各种问题，数据不在范围内了。

——written by LyonLys