【poj3415-长度不小于k的公共子串个数】后缀数组+单调栈

这题曾经用sam打过，现在学sa再来做一遍。

基本思路：计算A所有的后缀和B所有后缀之间的最长公共前缀。

分组之后，假设现在是做B的后缀。前面的串能和当前的B后缀产生的公共前缀必定是从前往后单调递增的，每次与h[i]取min时必定将栈尾一些长的全部取出来，搞成一个短的。

所以就开一个栈，栈里存的是长度，同时存一下它的出现此处cnt。

注意各种细节啊。。

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 using namespace std;

 typedef long long LL;
 const int N=*;
 int K,sl,cl,sa[N],rk[N],Rs[N],wr[N],y[N],h[N];
 LL sk[N],cnt[N];
 char s[N],c[N];

 void get_sa(int m)
 {
     for(int i=;i<=cl;i++) rk[i]=c[i];
     for(int i=;i<=m;i++) Rs[i]=;
     for(int i=;i<=cl;i++) Rs[rk[i]]++;
     for(int i=;i<=m;i++) Rs[i]+=Rs[i-];
     for(int i=cl;i>=;i--) sa[Rs[rk[i]]--]=i;

     int ln=,p=;
     while(p<cl)
     {
         int k=;
         for(int i=cl-ln+;i<=cl;i++) y[++k]=i;
         for(int i=;i<=cl;i++) if(sa[i]>ln) y[++k]=sa[i]-ln;

         for(int i=;i<=cl;i++) wr[i]=rk[y[i]];
         for(int i=;i<=m;i++) Rs[i]=;
         for(int i=;i<=cl;i++) Rs[wr[i]]++;
         for(int i=;i<=m;i++) Rs[i]+=Rs[i-];
         for(int i=cl;i>=;i--) sa[Rs[wr[i]]--]=y[i];

         for(int i=;i<=cl;i++) wr[i]=rk[i];
         for(int i=cl+;i<=cl+ln;i++) wr[i]=;
         p=;rk[sa[]]=;
         for(int i=;i<=cl;i++)
         {
             if(wr[sa[i]]!=wr[sa[i-]] || wr[sa[i]+ln]!=wr[sa[i-]+ln]) p++;
             rk[sa[i]]=p;
         }
         ln*=,m=p;
     }
     sa[]=,rk[]=;
 }

 void get_h()
 {
     int k=,j;
     for(int i=;i<=cl;i++) if(rk[i]!=)
     {
         j=sa[rk[i]-];
         if(k) k--;
         while(c[i+k]==c[j+k] && i+k<=cl && j+k<=cl) k++;
         h[rk[i]]=k;
     }
     h[]=;
 }

 void init()
 {
     int i,tl;cl=;
     scanf("%s",s+);
     tl=strlen(s+);sl=tl;
     for(i=;i<=sl;i++) c[++cl]=s[i];
     scanf("%s",s+);
     tl=strlen(s+);
     c[++cl]='#';
     for(i=;i<=sl;i++) c[++cl]=s[i];
 }

 bool check(int x,int tmp)
 {
     if(tmp==) return (x<=sl) ? :;
     else       return (x<=sl) ? :;
 }

 LL solve(int tmp)
 {
     int tot=;
     LL sum=,ans=;
     memset(sk,,sizeof(sk));
     memset(cnt,,sizeof(cnt));
     for(int i=;i<=cl;i++)
     {
         if(h[i]<K)
         {
             for(int j=;j<=tot;j++) cnt[j]=;
             tot=;sum=;
         }
         else
         {
             int tcnt=,tsum=;
             while(sk[tot] > h[i]-K+)
             {
                 tcnt+=cnt[tot];
                 tsum+=cnt[tot]*sk[tot];
                 sk[tot]=,cnt[tot]=;
                 tot--;
             }
             if(tcnt)
             {
                 sk[++tot]=h[i]-K+;
                 cnt[tot]=tcnt;
                 sum=sum-tsum+tcnt*sk[tot];
             }
             if(check(sa[i],tmp)) ans+=sum;
         }
         if(!check(sa[i],tmp) && (cl-sa[i]+>=K))
         {
             sk[++tot]=(cl-sa[i]+)-K+;
             cnt[tot]++;
             sum+=sk[tot];
         }
     }
     return ans;
 }

 int main()
 {
     freopen("a.in","r",stdin);
     freopen("me.out","w",stdout);
     while()
     {
         scanf("%d",&K);
         if(!K) return ;
         init();
         get_sa();
         get_h();
         // for(int i=1;i<=cl;i++) printf("%d ",sa[i]);printf("\n");
         // for(int i=1;i<=cl;i++) printf("%d ",rk[i]);printf("\n");
         // for(int i=1;i<=cl;i++)
         // {
             // for(int j=sa[i];j<=cl;j++) printf("%c",c[j]);printf("\n");
         // }
         printf("%I64d\n",solve()+solve());
     }
     return ;
 }