HDU-6230 Palindrome

最新推荐文章于 2020-09-09 12:51:43 发布

_zlWang

最新推荐文章于 2020-09-09 12:51:43 发布

阅读量248

点赞数

本文链接：https://blog.csdn.net/wl16wzl/article/details/81164248

版权

题目中要求的串，可以概括为两个回文串的组合：_____ i ____ j _____

以 i 和以 j 为中心的两个回文串拼接起来，注意 i 的回文串范围要覆盖 j，j 的回文串范围要覆盖 i

设p[i]是以 i 为中心的回文串覆盖半径（不包括 i ）

那么要符合三个条件：j>i ; j<=i+p[i] ; i>=j-p[j] ;

先用Manacher算法算出以每个点为中心的最长回文串长度

再按照 i 从1~n维护满足( j-p[j]>=i )的 j 在树状数组里，然后查询[ i+1,i+p[i] ]范围内的 j 有多少个，加起来就是答案

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn=500010;
ll c[maxn];
int n;
inline int lowbit(int x) {return x & -x;}
void update(int idx, int v) 
{
    while (idx <= n) 
    {
        c[idx] += v;
        idx += lowbit(idx);
    }
}
ll query(int x) // The sum of 1 to x
{
    ll ans = 0;
    while(x > 0) 
    {
        ans += c[x];
        x -= lowbit(x);
    }
    return ans;
}
char str[maxn];//原字符串
char tmp[maxn<<1];//转换后的字符串
int Len[maxn<<1];
int INIT(char *st)
{
    int i,len=strlen(st);
    tmp[0]='@';//字符串开头增加一个特殊字符，防止越界
    for(i=1;i<=2*len;i+=2)
    {
        tmp[i]='#';
        tmp[i+1]=st[i/2];
    }
    tmp[2*len+1]='#';
    tmp[2*len+2]='$';//字符串结尾加一个字符，防止越界
    tmp[2*len+3]=0;
    return 2*len+1;//返回转换字符串的长度
}
//Manacher算法计算过程
int MANACHER(char *st,int len)
{
     int mx=0,ans=0,po=0;//mx即为当前计算回文串最右边字符的最大值
     for(int i=1;i<=len;i++)
     {
         if(mx>i)
         Len[i]=min(mx-i,Len[2*po-i]);//在Len[j]和mx-i中取个小
         else
         Len[i]=1;//如果i>=mx，要从头开始匹配
         while(st[i-Len[i]]==st[i+Len[i]])
         Len[i]++;
         if(Len[i]+i>mx)//若新计算的回文串右端点位置大于mx，要更新po和mx的值
         {
             mx=Len[i]+i;
             po=i;
         }
         ans=max(ans,Len[i]);
     }
     return ans-1;//返回Len[i]中的最大值-1即为原串的最长回文子串额长度 
}
vector<int> g[maxn];
int p[maxn];
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%s",str);
		n=strlen(str);
		int len0=INIT(str);
		MANACHER(tmp,len0);
		memset(c,0,sizeof(c));
		for(int i=1;i<=n;i++) g[i].clear();
		for(int i=2;i<len0;i+=2)
		{
			p[i/2]=Len[i]/2-1;
			g[i/2-p[i/2]].push_back(i/2);
		}
		ll ans=0;
		for(int i=1;i<=n;i++)
		{
			for(int j=0;j<g[i].size();j++)
				update(g[i][j],1);
			ans+=query(i+p[i])-query(i);
		}
		printf("%lld\n",ans);
	}
	return 0;
}