【知识总结】回文自动机(Palindrome_Automaton)

参考资料:Palindromic Tree——回文树【处理一类回文串问题的强力工具】(请注意,其中似乎有一些错误)

回文自动机似乎和回文树是同一个东西qwq?

回文自动机(PAM)是一种处理回文串的工具。它的每个结点表示一个本质不同的回文串,转移边\(c\)表示在当前字符串的首尾分别加一个字符\(c\)

回文自动机由两棵树组成,根结点分别称为\(odd\)\(even\)\(even\)表示空串,长度为\(0\),长度为偶数的回文串在它的子树上;\(odd\)表示一个“虚拟”的串,长度为\(-1\),长度为奇数的回文串在它的子树上。\(odd\)的直接儿子表示只有一个字母的回文串。沿着转移边\(c\)走一步就在当前串首尾各加上一个字符\(c\)。和AC自动机类似,一个结点的\(fail\)指针指向它的最长回文真后缀(定义\(fail[even]=odd\))。比如wqwqqwqwq的回文自动机长这样(数字表示结点编号,红箭头表示\(fail\)指针):

wqwqqwqwq

我画着画着发现这个字符串里回文串比想象的多

和后缀自动机类似,构造回文自动机也采用每次插入一个字符的方法。设原串是\(S\),当前位置是\(pos\),要加入的字符是\(c\),则可能会多一些以字符\(c\)结尾的回文串。而多的这些字符串可以看成是一个回文串\([a,pos-1]\)满足\(S_{a-1}=c\)后面加一个字符\(c\)。于是要找到最长的这样的回文串\([a,pos-1]\),即从\(pos-1\)这个结点开始爬\(fail\)链,直到\(p\)点满足\(S_{pos-len[p]-1}=S_{pos}\)。爬\(fail\)链最终会到长度为\(-1\)\(even\),由于\(pos-(-1)-1=pos\),所以这个式子最终一定会成立。这个过程即代码中的\(get\_fail\)函数。

设第一个满足如上条件的点是\(p\)。如果\(p\)已经有了\(c\)字符的转移,则直接增加它的\(cnt\)(该字符串出现次数)即可;如果没有,则新建结点\(q\)\(q\)的长度显然是\(p\)的长度加\(2\)\(q\)\(fail\)是从\(p\)\(fail\)往上爬,找到第一个在后面加字符\(c\)仍为回文串的地方(方法同上述\(get\_fail\)),把它加字符\(c\)后转移到的点作为\(q\)\(fail\)

注意,如果要统计每个回文串的出现次数(即\(cnt\)),建完后要在\(fail\)树上做一遍树上递推(因为插入的时候只在当前点结尾的最长回文串的结点\(cnt\)上加\(1\)。如果一个串出现了,它的最长回文真后缀一定也出现了)。由于回文自动机是两棵树,所以不需要像后缀自动机求\(Right\)集合大小一样拓扑排序,只要按标号从大到小做即可。

题目:洛谷3649

把每个结点的长度\(len\)乘上出现次数\(cnt\)然后加起来就好了。

代码:

#include <cstdio>
#include <algorithm>
#include <cctype>
#include <cstring>
#include <string>
#define _ 0
using namespace std;

namespace zyt
{
	const int N = 3e5 + 10;
	template<typename T>
	inline bool read(T &x)
	{
		char c;
		bool f = false;
		x = 0;
		do
			c = getchar();
		while (c != EOF && c != '-' && !isdigit(c));
		if (c == EOF)
			return false;
		if (c == '-')
			f = true, c = getchar();
		do
			c = getchar();
		while (isdigit(c));
		if (f)
			x = -x;
		return true;
	}
	inline bool read(string &s)
	{
		static char buf[N];
		if (scanf("%s", buf) == -1)
			return false;
		else
		{
			s = buf;
			return true;
		}
	}
	template<typename T>
	inline void write(T x)
	{
		static char buf[20];
		char *pos = buf;
		if (x < 0)
			putchar('-'), x = -x;
		do
			*pos++ = x % 10 + '0';
		while (x /= 10);
		while (pos > buf)
			putchar(*--pos);
	}
	const int CH = 26;
	typedef long long ll;
	string s;
	namespace Palindrome_Auto_Machine
	{
		struct node
		{
			int len, cnt, fail, s[CH];
		}tree[N];
		int cnt, last, odd, even, pos;
		char s[N];
		void init()
		{
			last = even = 0, odd = 1, cnt = 1, pos = 0;
			s[0] = '#';
			tree[odd].len = -1, tree[even].len = 0;
			tree[odd].fail = tree[even].fail = odd;
		}
		int get_fail(int p)
		{
			while (s[pos - tree[p].len - 1] != s[pos])
				p = tree[p].fail;
			return p;
		}
		void insert(const char c)
		{
			s[++pos] = c;
			int x = c - 'a', p = get_fail(last);
			if (!tree[p].s[x])
			{
				tree[++cnt].len = tree[p].len + 2;
				tree[cnt].fail = tree[get_fail(tree[p].fail)].s[x];
				tree[p].s[x] = cnt;
			}
			last = tree[p].s[x];
			++tree[last].cnt;
		}
		void build(const string &str)
		{
			for (int i = 0; i < str.size(); i++)
				insert(str[i]);
			for (int i = cnt; i > 0; i--)
				tree[tree[i].fail].cnt += tree[i].cnt;
		}
		inline ll solve()
		{
			ll ans = 0;
			for (int i = 0; i <= cnt; i++)
				ans = max(ans, (ll)tree[i].cnt * tree[i].len);
			return ans;
		}
	}
	int work()
	{
		using Palindrome_Auto_Machine::init;
		using Palindrome_Auto_Machine::build;
		using Palindrome_Auto_Machine::solve;
		read(s);
		init();
		build(s);
		write(solve());
		return (0^_^0);
	}
}
int main()
{
	return zyt::work();
}
posted @ 2019-01-26 19:49  Inspector_Javert  阅读(620)  评论(0编辑  收藏  举报