【CF245H】Queries for Number of Palindromes（回文树）

【CF245H】Queries for Number of Palindromes（回文树）

题面

洛谷

题解

回文树，很类似原来一道后缀自动机的题目
后缀自动机那道题
看到(n)的范围很小，但是(Query)很多
所以提前预处理出每一段(l,r)的答案
时间复杂度(O(n^2+Q))

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 5050
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
int ans[MAX][MAX];
char s[MAX];
char ch[MAX];
struct PT
{
	struct Node
	{
		int son[26];
		int len,ff;
	}t[MAX];
	int last,tot,dep[MAX];
	void init()
	{
		t[tot=1].len=-1;
		t[last=0].ff=t[1].ff=1;
	}
	void clear()
	{
		memset(t,0,sizeof(t));
		memset(dep,0,sizeof(dep));
		init();
	}
	void extend(int c,int n,char *s)
	{
		int p=last;
		while(s[n-t[p].len-1]!=s[n])p=t[p].ff;
		if(!t[p].son[c])
		{
			int v=++tot,k=t[p].ff;
			while(s[n-t[k].len-1]!=s[n])k=t[k].ff;
			t[v].ff=t[k].son[c];
			t[v].len=t[p].len+2;
			t[p].son[c]=v;
			dep[v]=dep[t[v].ff]+1;
		}
		last=t[p].son[c];
	}
}PT;
int main()
{
	scanf("%s",s+1);
	int n=strlen(s+1);
	for(int i=1;i<=n;++i)
	{
		PT.clear();
		for(int j=i;j<=n;++j)ch[j-i+1]=s[j];
		for(int j=i;j<=n;++j)
		{
			PT.extend(ch[j-i+1]-97,j-i+1,ch);
			ans[i][j]=ans[i][j-1]+PT.dep[PT.last];
		}
	}
	int Q=read();
	while(Q--)
	{
		int l=read(),r=read();
		printf("%d
",ans[l][r]);
	}
	return 0;
}