【回文串】Codeforces 17E Palisection

点击打开链接

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

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


#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
#include 
#include 
#include 
#include 
#include 
#include 
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include 
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<


你可能感兴趣的:(字符串)