【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\n",ans[l][r]);

	}

	return 0;

}