HDU 6194 string string string ——（2017沈阳网络赛，后缀数组）

　　思路见：http://blog.csdn.net/aozil_yang/article/details/77929216。

　　代码如下：

 #include <stdio.h>
 #include <algorithm>
 #include <string.h>
 using namespace std;
 const int N = 1e5 + ;
 typedef long long ll;

 /**
  *    sa[i]:表示排在第i位的后缀的起始下标
  *    rank[i]:表示后缀suffix(i)排在第几
  *    height[i]:sa[i-1] 与 sa[i]的LCP(最长公共前缀)值
  *
  * */
  /*
     如果整数的话模板改成int.
     加一个数a[n] = 0 。 这样他的排名是第一个。
     construct(a,n+1);

     字符串的话。
     len = strlen(str);
     construct(s,strlen(s)+1);
     排名第0的是个空字符串。

     height[i]:sa[i-1] 与 sa[i]的LCP(最长公共前缀)值
     所以height[1] = 0;
     rank[len] = 0;
     sa[0] = len;
  */
 int sa[N],rnk[N],height[N];
 void construct(const char *s,int n,int m = ) {
     static int t1[N],t2[N],c[N];
     int *x = t1,*y = t2;
     int i,j,k,p,l;
     for (i = ; i < m; ++ i) c[i] = ;
     for (i = ; i < n; ++ i) c[x[i] = s[i]] ++;
     for (i = ; i < m; ++ i) c[i] += c[i - ];
     for (i = n - ; i >= ; -- i) sa[--c[x[i]]] = i;
     for (k = ; k <= n; k <<= ) {
         p = ;
         for (i = n - k; i < n; ++ i) y[p++] = i;
         for (i = ; i < n; ++ i) if (sa[i] >= k) y[p++] = sa[i] - k;
         for (i = ; i < m; ++ i) c[i] = ;
         for (i = ; i < n; ++ i) c[x[y[i]]] ++;
         for (i = ; i < m; ++ i) c[i] += c[i - ];
         for (i = n - ; i >= ; -- i) sa[--c[x[y[i]]]] = y[i];
         std::swap(x,y);
         p = ; x[sa[]] = ;
         for (i = ; i < n; ++ i)
             x[sa[i]] = y[sa[i - ]] == y[sa[i]]
                 && y[sa[i - ] + k] == y[sa[i] + k] ? p - : p ++;
         if (p >= n) break;
         m = p;
     }
     for (i = ; i < n; ++ i) rnk[sa[i]] = i;
     for (i = ,l = ; i < n; ++ i) {
         if (rnk[i]) {
             j = sa[rnk[i] - ];
             while (s[i + l] == s[j + l]) l++;
             height[rnk[i]] = l;
             if (l) l--;
         }
     }
 }

 int st_min[N][];
 int preLog[N];
 int n;
 void init_rmq()
 {
     preLog[] = ;
     for(int i=;i<=n;i++)
     {
         preLog[i] = preLog[i-];
         if(( << preLog[i] + ) == i) preLog[i]++;
     }
     for(int i=n;i>=;i--)
     {
         st_min[i][] = height[i];
         for(int j=;(i+(<<j)-)<=n;j++)
         {
             st_min[i][j] = min(st_min[i][j-], st_min[i+(<<j-)][j-]);
         }
     }
 }
 int query_min(int l,int r)
 {
     int len = r - l + , k = preLog[len];
     return min(st_min[l][k], st_min[r-(<<k)+][k]);
 }
 char s[N];
 int lcp(int l, int r)
 {
     /*int ql = rnk[l], qr = rnk[r];
     if(ql > qr) swap(ql, qr);
     if(ql == qr) return n - l;
     return query_min(ql+1, qr);*/
     // l and r are number in rnk.
     if(l == r) return n - sa[l];
     return query_min(l+, r);
 }

 int main()
 {
     int T; scanf("%d",&T);
     while(T--)
     {
         int k; scanf("%d",&k);
         scanf("%s",s);
         n = strlen(s);
         s[n] = ;
         construct(s, n+);
         init_rmq();

         ll ans = ;
         for(int i=;i+k-<=n;i++)
         {
             ans += lcp(i, i+k-);
             if(i- >= ) ans -= lcp(i-, i+k-);
             if(i+k <= n) ans -= lcp(i, i+k);
             if(i- >=  && i+k <= n) ans += lcp(i-, i+k);
         }
         printf("%I64d\n",ans);
     }
     return ;
 }

　　

　　最后仍然需要注意的是我sa的模板中，除了rnk数组其他都是0-base的。