1. 程式人生 > 其它 >luogu P5829 【模板】失配樹

luogu P5829 【模板】失配樹

傳送門


首先,一個串的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;
}