bzoj2534: Uva10829L-gap字符串

Description

有一种形如uvu形式的字符串,其中u是非空字符串,且V的长度正好为L,那么称这个字符串为L-Gap字符串
给出一个字符串S,以及一个正整数L,问S中有多少个L-Gap子串.

Input

第一行一个数字L
第二行一个字符串S

Output

一个数字表示S中有多少个L-Gap子串.

Sample Input

3
aabbaa

Sample Output

2

Hint

S的长度不超过50000,L<=10

 

题解：http://www.cnblogs.com/iamCYY/p/4730777.html

Orz CYY

 

code：

 #include<cstdio>
 #include<iostream>
 #include<cmath>
 #include<cstring>
 #include<algorithm>
 #define maxn 50050
 using namespace std;
 char ch,s[maxn];
 int n,l,ans;
 int fa[maxn],root[maxn],list[maxn],cnt,que[maxn];
 int SA[maxn],rank[maxn],height[maxn],sum[maxn],t1[maxn],t2[maxn];
 int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
 struct Splay{
     int tot,son[maxn][],fa[maxn],siz[maxn],val[maxn];
     void init(){
         for (int i=;i<=n;i++) siz[++tot]=;
         for (int i=;i<=n;i++) val[i]=SA[i];
     }
     int which(int x){return son[fa[x]][]==x;}
     void update(int x){siz[x]=siz[son[x][]]+siz[son[x][]]+;}
     void rotate(int x){
         int y=fa[x],d=which(x),dd=which(y);
         fa[son[x][d^]]=y,son[y][d]=son[x][d^],fa[x]=fa[y];
         if (fa[x]) son[fa[x]][dd]=x;
         son[x][d^]=y,fa[y]=x,update(y),update(x);
     }
     void splay(int x){
         while (fa[x]){
             if (!fa[fa[x]]) rotate(x);
             else if (which(x)==which(fa[x])) rotate(fa[x]),rotate(x);
             else rotate(x),rotate(x);
         }
         root[find(x)]=x;
     }
     void insert(int root,int x){
         int f,t;
         for (f=t=root;t;t=son[t][val[x]>val[t]]) f=t;
         son[f][val[x]>val[f]]=x,fa[x]=f,splay(x);
     }
     void merge(int x){while (cnt) insert(root[x],que[cnt--]);}
     int find_pre(int root,int v){
         int x,t=;
         for (x=root;x;val[x]<v?t=x,x=son[x][]:x=son[x][]);
         if (t) splay(t);
         return t?siz[son[t][]]+:;
     }
     int find_next(int root,int v){
         int x,t=;
         for (x=root;x;val[x]>v?t=x,x=son[x][]:x=son[x][]);
         if (t) splay(t);
         return t?siz[son[t][]]+:;
     }
     void get(int x){
         if (son[x][]) get(son[x][]);
         que[++cnt]=x;
         if (son[x][]) get(son[x][]);
         son[x][]=son[x][]=fa[x]=;
     }
 }T;
 bool ok;
 void read(int &x){
     for (ok=,ch=getchar();!isdigit(ch);ch=getchar()) if (ch=='-') ok=;
     for (x=;isdigit(ch);x=x*+ch-'',ch=getchar());
     if (ok) x=-x;
 }
 bool cmp(int x,int y){
     if (height[x]!=height[y]) return height[x]>height[y];
     return x<y;
 }
 void get_SA(){
     int *x=t1,*y=t2,m=,tot=;
     for (int i=;i<=n;i++) sum[x[i]=s[i]]++;
     for (int i=;i<=m;i++) sum[i]+=sum[i-];
     for (int i=n;i>=;i--) SA[sum[x[i]]--]=i;
     for (int len=;tot<n;len<<=,m=tot){
         tot=;
         for (int i=n-len+;i<=n;i++) y[++tot]=i;
         for (int i=;i<=n;i++) if (SA[i]>len) y[++tot]=SA[i]-len;
         for (int i=;i<=m;i++) sum[i]=;
         for (int i=;i<=n;i++) sum[x[y[i]]]++;
         for (int i=;i<=m;i++) sum[i]+=sum[i-];
         for (int i=n;i>=;i--) SA[sum[x[y[i]]]--]=y[i];
         swap(x,y),x[SA[]]=tot=;
         for (int i=;i<=n;i++){
             if (y[SA[i]]!=y[SA[i-]]||y[SA[i]+len]!=y[SA[i-]+len]) tot++;
             x[SA[i]]=tot;
         }
     }
     for (int i=;i<=n;i++) rank[i]=x[i];
 }
 void get_height(){
     for (int i=,j=;i<=n;i++){
         if (rank[i]==) continue;
         while (s[i+j]==s[SA[rank[i]-]+j]) j++;
         height[rank[i]]=j;
         if (j>) j--;
     }
 }
 int main(){
     read(l),scanf("%s",s+),n=strlen(s+);
     get_SA(),get_height(),T.init();
     for (int i=;i<=n;i++) root[i]=i;
     for (int i=;i<=n;i++) fa[i]=i;
     for (int i=;i<n;i++) list[i]=i+;
     sort(list+,list+n,cmp);
     for (int i=;i<n;i++){
         int x=find(list[i]-),y=find(list[i]),lcp=height[list[i]]; cnt=;
         if (T.siz[root[x]]>T.siz[root[y]]) swap(x,y);
         int tsiz=T.siz[root[y]];
         T.get(root[x]);
         for (int j=,t=que[j];j<=cnt;j++,t=que[j]){
             ans+=T.find_next(root[y],SA[t]+l);
             ans+=T.find_pre(root[y],lcp+SA[t]+l+)-tsiz;
             ans+=T.find_next(root[y],SA[t]-lcp-l-);
             ans+=T.find_pre(root[y],SA[t]-l)-tsiz;
         }
         fa[x]=y,T.merge(y);
     }
     printf("%d\n",ans);
     return ;
 }