hdu5157 不交叉回文对统计

最新推荐文章于 2020-10-12 13:34:03 发布

lshacm

最新推荐文章于 2020-10-12 13:34:03 发布

阅读量399

点赞数

分类专栏： acm_数据结构

本文链接：https://blog.csdn.net/u013514182/article/details/42381251

版权

acm_数据结构专栏收录该内容

39 篇文章 1 订阅

订阅专栏

关键词：manacher算法、统计以第i个字符为起点/终点的回文串个数！

将新数组的元素扫描一遍，标记需要更新点的起始和末了位置，起点++，终点--，最后统计以第i个字符为起点的数量时，将前i个累加起来即可。因为在[起点,终点]区间的元素都加上了起点的那个1，不在[起点,终点]区间的元素由于终点-1，因此没有加到终点的那个1。本题需要考虑奇偶分类和边界调节考虑等，十分繁杂，代码无力Orz...

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<map>
#include<vector>
#include<queue>
#include<stack>
#define ll long long
#define sf scanf
#define pf printf
#define INF 0x3f3f3f3f
#define maxn 110010
#define mem(a,b) memset(a,b,sizeof(a))
#define lowbit(x) x&(-x)
const ll mol=1000000007;
using namespace std;

char s[maxn<<2],str[maxn<<2];
int p[maxn<<2];
ll cntr[maxn<<2],cntl[maxn<<2],ans;
ll pre[maxn<<2],suf[maxn<<2];
int l;

void segadd(ll s[],int l,int r){
    s[l]++,s[r+1]--;
}

void turn(ll s[],int n){
    for(int i=1;i<=n;i++) s[i]+=s[i-1];
}

int Min(int a,int b){
    return (a>b);
}

void work(){
    memset(cntr,0,sizeof(cntr));
    memset(cntl,0,sizeof(cntl));
    ans=0;
    for(int i=0;i<2*l+1;i++){
        int L=(i-p[i]+1)/2+1,r=(i+p[i]-1)/2-1+1,lm=(i-1)/2+1,rm=i/2+1;//找规律...纯属抄袭...实在太难( ⊙ o ⊙ )啊！
        segadd(cntr,rm,r);
        segadd(cntl,L,lm);
    }
    turn(cntr,l);
    turn(cntl,l);
    pre[0]=0;
    for(int i=1;i<=l;i++) pre[i]=pre[i-1]+cntr[i];
    ans=0;
    for(int i=1;i<l;i++){
        ans+=pre[i]*cntl[i+1];
    }
    printf("%I64d\n",ans);
    return;
}

void Manacher(char S[],int N,int P[]){
    int mx=1,id=0;
    P[0]=1;
    for(int i=1;i<N;i++){
        if(mx>i) P[i]=min(P[2*id-i],mx-i);
        else P[i]=1;
        while(P[i]<=i&&S[i-P[i]]==S[i+P[i]]) P[i]++;
        if(i+P[i]>mx){
            mx=i+P[i];
            id=i;
        }
    }
}

int main(){
    while(scanf("%s",s)!=EOF){
        l=strlen(s);
        mem(p,0),mem(str,0);
        str[0]='#';
        for(int i=0;i<l;i++){ str[2*i+1]=s[i],str[2*i+2]='#'; }
        str[2*l+1]='\0';
        Manacher(str,2*l+1,p);
        work();
    }
    return 0;
}