【回文串】Codeforces 17E Palisection

最新推荐文章于 2020-01-15 09:23:35 发布

Kewowlo

最新推荐文章于 2020-01-15 09:23:35 发布

阅读量798

点赞数

分类专栏：字符串

本文链接：https://blog.csdn.net/kewowlo/article/details/48471919

版权

字符串专栏收录该内容

7 篇文章 0 订阅

订阅专栏

点击打开链接

题意：计算相交的回文子串对数

先计算出总的数量减去不相交的回文子串的对数

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
#include <queue>
#include <stack>
#include <vector>
#include <bitset>
#include <deque>
#include <set>
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <map>
typedef long long LL;
const int INF = 1<<10;
const LL mod = 51123987;
const int MAXN = 2000001;
char s[MAXN*2],str[MAXN*2];
int p[MAXN*2],len;
LL num1[MAXN],num2[MAXN],num3[MAXN];
int manacher()
{
    int n=0;
    for(int i=1;i<=len;i++)
    {
        num1[i]=num2[i]=num3[i]=0;
        s[++n]='#';
        s[++n]=str[i];
    }
    s[++n]='#';
    int mx=0,id=0;
    for(int i=1;i<=n;i++)
    {
        if(mx>=i) p[i]=min(p[2*id-i],mx-i);
        else p[i]=0;
        while(i-p[i]-1>=1&&i+p[i]+1<=n&&s[i+p[i]+1]==s[i-p[i]-1]) p[i]++;
        if(i+p[i]>mx)
            mx=i+p[i],id=i;
    }
    for(int i=1;i<=len;i++)
    {
        num1[i-p[2*i]/2]++;//num1[i] 表示 i这位置起始的回文串数量
        num1[i+1]--;

        num1[i-p[2*i+1]/2+1]++;
        num1[i+1]--;

        num2[i]++;//num2[i] 表示 i这位置结束的回文串数量
        num2[i+p[2*i]/2+1]--;
        
        num2[i+1]++;
        num2[i+p[2*i+1]/2+1]--;
    }
    for(int i=1;i<=len;i++) num1[i]+=num1[i-1],num2[i]+=num2[i-1];
    for(int i=1;i<=len;i++) num3[i]=(num3[i-1]+num2[i])%mod;
    LL ans=num3[len]*(num3[len]-1)/2%mod;//相交的回文串总数
    for(int i=1;i<=len;i++)//
        ans=(ans-num1[i]*num3[i-1]+mod)%mod;
    ans=(ans+mod)%mod;
    return ans;
}
int main()
{
    scanf("%d %s",&len,str+1);
    cout<<manacher()<<endl;
    return 0;
}