后缀自动机(SAM)-模板记录/好文推荐

记录一个菜逼的成长。。

好文推荐
个人感觉通俗易懂。

这里已hihocoder第128周为例。
hihocoder第128周
上面也有讲解。

题意:一个字符串里有几种不同的子串
答案:SAM所有状态的字符串数的和就是。每个状态的字符串数 = (max(s) - min(s) + 1)
max(s):是状态s所能表示的最大字符串长度
min(s)同理。

这里记下简易模板。

#include <bits/stdc++.h>
using namespace std; 
#define fi first
#define se second
typedef long long LL;
const int maxn = 1000000 + 10;
struct SAM{
  int link[maxn<<1],go[maxn<<1][26],mxlen[maxn<<1],mnlen[maxn<<1];
  int tot,last;
  void init(){
    tot = last = 0;
    memset(go[0],-1,sizeof(go[0]));
    link[0] = -1; mxlen[0] = 0;mnlen[0] = 0;
  }
  void ins(int c){
    int p = last,cur = ++tot;
    mxlen[cur] = mxlen[p] + 1;
    memset(go[cur],-1,sizeof(go[cur])); 
    while(p != -1 && go[p][c] == -1)go[p][c] = cur,p = link[p];
    if (p == -1) {
      mnlen[cur] = 1;
      link[cur] = 0;
    }
    else {
      int q = go[p][c];
      if (mxlen[q] == mxlen[p] + 1) mnlen[cur] = mxlen[q] + 1,link[cur] = q;
      else {
        int nq = ++tot;
        memcpy(go[nq],go[q],sizeof(go[q]));
        mxlen[nq] = mxlen[p] + 1;
        link[nq] = link[q]; mnlen[q] = mxlen[nq] + 1;
        link[cur] = nq; mnlen[cur] = mxlen[nq] + 1;
        link[q] = nq;
        while(p != -1 && go[p][c] == q) {
          go[p][c] = nq,p = link[p];
        }
        mnlen[nq] = mxlen[link[nq]] + 1;
      }
    }
    last = cur;
  }
}sam;
char str[maxn];
int main()
{
  while(~scanf("%s",str)) {
    sam.init();
    for (int i = 0; str[i]; i++)sam.ins(str[i] - 'a');
    LL ans = 0;
    for (int i = 1; i <= sam.tot; i++) {
      ans += sam.mxlen[i] - sam.mnlen[i] + 1;
    }
    printf("%lld\n",ans);
  }
  return 0;
}