HDU-5785 Interesting(Manacher算法+区间处理)

题目大意:给一个字符串,求所有相邻两回文子串的外侧下标之积的和

题目分析:另L[i]为所有以 i 为右端点的回文字串的左端点之和,同理,另R[i]表示所有以 i 为左端点的回文子串的右端点之和。显然,答案为sigma(L[i]*R[i+1]) 其中,1<=i<length(字符串)。求出L和R是关键。先用manacher算法处理出p数组,然后再求出L和R。求L和R的思想(非常巧妙)跟树状数组求区间和的思想差不多。不过,这道题如果用树状数组或线段树的话会超时。

参考代码:

# include<iostream>
# include<cstdio>
# include<cstring>
# include<algorithm>
using namespace std;
# define LL long long const int mod=1000000007;
const int N=1000000; char str[N+5];
char m[(N<<1)+5];
int p[(N<<1)+5]; LL addtag[2][N+5];
LL subtag[2][N+5];
LL addcnt[2][N+5];
LL subcnt[2][N+5];
LL val[2][N+5]; void manacher(char *ch)
{
int len=strlen(ch+1);
int cnt=0;
m[cnt++]='@';
m[cnt++]='#';
for(int i=1;i<=len;++i){
m[cnt++]=ch[i];
m[cnt++]='#';
}
m[cnt]='\0';
int id,mx=0;
for(int i=1;i<cnt;++i){
if(mx>i) p[i]=min(mx-i,p[2*id-i]);
else p[i]=1;
while(m[i-p[i]]==m[i+p[i]]) ++p[i];
if(i+p[i]>mx){
mx=i+p[i];
id=i;
}
}
} void update(int id,int l,int r,int a)
{
addtag[id][l]+=a;
addtag[id][l]%=mod;
++addcnt[id][l];
subtag[id][r+1]+=a-r+l;
subtag[id][r+1]%=mod;
++subcnt[id][r];
} void pushDown(int n)
{
for(int i=1;i<=n;++i){
if(addcnt[0][i]){
val[0][i]+=addtag[0][i];
val[0][i]%=mod;
addtag[0][i+1]+=addtag[0][i]-addcnt[0][i];
addtag[0][i+1]%=mod;
addcnt[0][i+1]+=addcnt[0][i];
}
if(subcnt[0][i]){
val[0][i]-=subtag[0][i];
val[0][i]%=mod;
subtag[0][i+1]+=subtag[0][i]-subcnt[0][i];
subtag[0][i+1]%=mod;
subcnt[0][i+1]+=subcnt[0][i];
}
if(addcnt[1][i]){
val[1][i]+=addtag[1][i];
val[1][i]%=mod;
addtag[1][i+1]+=addtag[1][i]-addcnt[1][i];
addtag[1][i+1]%=mod;
addcnt[1][i+1]+=addcnt[1][i];
}
if(subcnt[1][i]){
val[1][i]-=subtag[1][i];
val[1][i]%=mod;
subtag[1][i+1]+=subtag[1][i]-subcnt[1][i];
subtag[1][i+1]%=mod;
subcnt[1][i+1]+=subcnt[1][i];
}
}
} int main()
{
while(~scanf("%s",str+1))
{
manacher(str);
memset(val,0,sizeof(val));
memset(addtag,0,sizeof(addtag));
memset(subtag,0,sizeof(subtag));
memset(addcnt,0,sizeof(addcnt));
memset(subcnt,0,sizeof(subcnt));
for(int i=1;m[i];++i){
if(i&1){
if(p[i]==1) continue;
update(0,i/2+1,i/2+p[i]/2,i/2);
update(1,i/2-p[i]/2+1,i/2,i/2+p[i]/2);
}else{
int id=i/2;
int l=(p[i]-1)/2;
update(0,id,id+l,id);
update(1,id-l,id,id+l);
}
}
int n=strlen(str+1);
pushDown(n);
LL ans=0;
for(int i=1;i<n;++i){
ans+=val[0][i]*val[1][i+1];
ans%=mod;
}
while(ans<0) ans+=mod;
printf("%lld\n",ans);
}
return 0;
}

  

上一篇:MarkDown+LaTex 数学内容编辑样例收集


下一篇:火狐和ie下获取javascript 获取event