hdu5157 不交叉回文对统计

来源:互联网 发布:全力以赴歌曲网络歌手 编辑:程序博客网 时间:2024/06/05 15:05

关键词: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;}


0 0
原创粉丝点击