SAM-模板和学习

先上一道裸题代码：给定若干个(<=10)由小写字母组成的字符串(每个字符串长度不超过10^5)，求他们的最长公共子串的长度。

用的是在每个字符串中间加间隔符，把它们合成一个字符串，最后 DFS + 标记 找答案的方法。

#include <stdio.h>
#include <unordered_map>
#include <string.h>
#include <vector>

using namespace std;

const int _N = 100005;

char str[_N];
vector<int> G[_N*20];

namespace SAM {
    int root, tot, last, lcs, par[_N*20], mx[_N*20], mk[_N*20];
    unordered_map<int, int> son[_N*20];
    
    int Ins(int v_mx, int v_mk) { mx[++tot] = v_mx; mk[tot] = v_mk; return tot; }
    
    void Init() { tot = 0; root = last = Ins(0, 0); return; }
    
    void dfs(int node, int FULL)
    {
        for (int i = G[node].size()-1; i >= 0; --i)
            dfs(G[node][i], FULL), mk[node] |= mk[G[node][i]];
        if (mk[node] == FULL && lcs < mx[node])
            lcs = mx[node];
    }
    
    void GetLCS(int FULL)
    {
        int lcs = 0, i;
        for (i = 1; i <= tot; ++i)
            G[par[i]].push_back(i);
        lcs = 0;
        dfs(root, FULL);
        return;
    }
    
    void Extend(int t, int id)
    {
        int np, nq, p, q;
        np = Ins(mx[last] + 1, id);//---
        for (p = last; p && !son[p][t]; p = par[p])
            son[p][t] = np;
        if (!p) {
            par[np] = root;
        } else {
            q = son[p][t];
            if (mx[q] == mx[p] + 1) {
                par[np] = q;
            } else {
                nq = Ins(mx[p] + 1, mk[q]);//---
                son[nq] = son[q];
                par[nq] = par[q], par[q] = par[np] = nq;
                for ( ; p && son[p][t] == q; p = par[p])
                    son[p][t] = nq;
            }
        }
        last = np;
        return;
    }
    
}

int main()
{
    int i, cnt = 1, len;
    scanf("%s", str);
    SAM::Init();
    len = strlen(str);
    for (i = 0; i < len; ++i)
        SAM::Extend(str[i]-'a', cnt);
    while (scanf("%s", str) != -1) {
        len = strlen(str);
        cnt <<= 1;
        SAM::Extend(26, cnt);
        for (i = 0; i < len; ++i)
            SAM::Extend(str[i]-'a', cnt);
    }
    SAM::GetLCS((cnt<<1)-1);
    printf("%d
", SAM::lcs);
    return 0;
}

其实现在也不太会 SAM OrzOrzOrz，好像给自己埋了一个坑， SAM 相关的训练也已经结束了，只能以后慢慢填啦。

这两个月打了不少比赛，没写题解，等放假补吧OrzOrz...