luogu P5829 【模板】失配樹
阿新 • • 發佈:2021-09-01
傳送門
首先,一個串的border定義為所有滿足其字首等於字尾的長度集合。
那怎麼求border呢?會發現上面的定義就是kmp中的\(fail[|S|]\),而所有的border就是沿著這個fail指標一直跳下去,即\(fail[|S|],fail[fail[|S|]], \cdots\)
那麼求兩個字首的最長公共border,就是從他們能跳到的所有長度中找公共最大的。如果我們從\(fail[i]\)向\(i\)連一條邊,就形成了一棵樹(失配樹),而最長公共border就是他們兩個的父親的lca(不是本身,因為border不能包含自己)。
於是這道板子題就做出來了,感覺挺好理解的。
#include<cstdio> #include<iostream> #include<cmath> #include<algorithm> #include<cstring> #include<cstdlib> #include<cctype> #include<vector> #include<queue> #include<assert.h> #include<ctime> using namespace std; #define enter puts("") #define space putchar(' ') #define Mem(a, x) memset(a, x, sizeof(a)) #define In inline #define forE(i, x, y) for(int i = head[x], y; ~i && (y = e[i].to); i = e[i].nxt) typedef long long ll; typedef double db; const int INF = 0x3f3f3f3f; const db eps = 1e-8; const int maxn = 1e6 + 5; const int N = 20; In ll read() { ll ans = 0; char ch = getchar(), las = ' '; while(!isdigit(ch)) las = ch, ch = getchar(); while(isdigit(ch)) ans = (ans << 1) + (ans << 3) + ch - '0', ch = getchar(); if(las == '-') ans = -ans; return ans; } In void write(ll x) { if(x < 0) x = -x, putchar('-'); if(x >= 10) write(x / 10); putchar(x % 10 + '0'); } int n, m; char s[maxn]; int f[maxn], dep[maxn], fa[N + 2][maxn]; In void init() { f[1] = 0, dep[1] = 1; for(int i = 2, j = 0; i <= n; ++i) { while(j && s[j + 1] != s[i]) j = f[j]; if(s[j + 1] == s[i]) ++j; f[i] = j; fa[0][i] = j, dep[i] = dep[j] + 1; } for(int i = 1; i <= n; ++i) for(int j = 1; j <= N; ++j) fa[j][i] = fa[j - 1][fa[j - 1][i]]; } In int lca(int x, int y) { if(dep[x] < dep[y]) swap(x, y); for(int i = N; i >= 0; --i) if(dep[fa[i][x]] >= dep[y]) x = fa[i][x]; if(x == y) return x; for(int i = N; i >= 0; --i) if(fa[i][x] ^ fa[i][y]) x = fa[i][x], y = fa[i][y]; return fa[0][x]; } int main() { scanf("%s", s + 1); n = strlen(s + 1); init(); m = read(); for(int i = 1; i <= m; ++i) { int x = read(), y = read(); write(lca(fa[0][x], fa[0][y])), enter; } return 0; }