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 = ;
 const int LAST = ;
 const int K = ;

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

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

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

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

         temp->Clear(d);

         return temp;
     }

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

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

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

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

                 *nq = *q;
                 nq->dist = p->dist + ;
                 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 << ];
     Node *ptr[N << ];

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

             p->sub = ;
             for (int j = ; j < K; ++j) {
                 if (p->nx[j]) {
                     p->sub += p->nx[j]->sub;
                 }
             }
         }
         --node[].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());
     for (int i = ; i < LAST; ++i) {
         s[i] = rand() %  + 'a';
     }
     s[LAST] = ;
 }

 int Run() {
     //while (~scanf("%s", s)) {
     //while (1) {
         //Generate(s);
         scanf("%s", s);
         sam.Clear();
         for (int i = ; 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 = ;

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

     return ;
 }

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

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

——written by LyonLys