【金牌导航】【Manacher】不相交回文串

【金牌导航】【Manacher】不相交回文串

题目

---

解题思路

用manacher求出回文半径
下文代码给注释
再得出以每位开始的回文串的数量pre，以及在每位之前（包括这一位）结束的回文串的数量sum
用差分来求pre和suf（sum求suf的前缀和，suf是在当前位结束的数量）
最后ans统计答案pre[i]*sum[i-1]

---

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
char s[100010],c[200010];
long long ans,lens,lenc,r[200010],sum[200010],suf[200010],pre[200010];
int main()
{
	while (scanf("%s",s)!=EOF)
	{
		  lens=strlen(s);
		  c[0]='$',c[1]='#';
		  lenc=1,ans=0;
		  long long mx=0,p=0;
		  memset(r,0,sizeof(r));
		  memset(sum,0,sizeof(sum));
		  memset(suf,0,sizeof(suf));
		  memset(pre,0,sizeof(pre));
		  for (int i=0;i<lens;i++)
		      c[++lenc]=s[i],c[++lenc]='#'; 
		  for (int i=0;i<lenc;i++)
		  {  
		  	  if (i<mx) r[i]=min(mx-i,r[p*2-i]);  //mx为回文串覆盖到的最右边界，p为回文中心，如果小于mx,说明i有关于p对称的j，即p-(i-p)=p*2-i,如果r[p*2-i]+i超过mx-i就取小的一个（因为不在回文串内了，不保证有回文）
		  	  while (c[i+r[i]]==c[i-r[i]]) r[i]++;  //暴力扩展
		  	  if (i+r[i]>mx)  //更新
		  	     mx=i+r[i],p=i;
		  } 
		  for (int i=2;i<=lens*2;i++)  //差分求pre和suf
		  {
		  	  mx=(i+1)/2;
		  	  suf[mx]++,suf[mx+r[i]/2]--; 
		  }
		  for (int i=lens*2;i>1;i--)
		  {
		  	  mx=i/2;
		  	  pre[mx]++,pre[mx-r[i]/2]--;
		  }
		  for (int i=1;i<=lens;i++)
		      suf[i]+=suf[i-1],sum[i]=sum[i-1]+suf[i];
		  for (int i=lens;i>0;i--) pre[i]+=pre[i+1];
		  for (int i=1;i<=lens;i++) ans+=sum[i-1]*pre[i];
		  printf("%lld\n",ans);
	}
	return 0;
}