hdu4691（后缀数组求最长公共前缀）

学习链接    http://tieba.baidu.com/f?kz=754580296
题意：题目的本质就是求输入的相邻的两个串的最长公共字串，有sa来解求得height数组，然后用rmq算法查询。

解题思路：sa模板  + ST模板

代码如下：

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<math.h>
#include<cstring>
#include<string>
#define N 100005
#define inf 0x3f3f3f3f
#define pi acos(-1.0)
#define eps 10e-6
using namespace std;
char s[N];
int wa[N],wb[N],wv[N],WS[N],r[N],sa[N];//sa的有效范围[0,n]
int cmp(int *r,int a,int b,int l)
{
	return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int n,int m)//n为字符串的长度加1，因为在使用倍增算法前在原字符串后面加一个0
{
    int i,j,p,*x=wa,*y=wb,*t;
    for(i=0;i<m;i++) WS[i]=0;
    for(i=0;i<n;i++) WS[x[i]=r[i]]++;
    for(i=1;i<m;i++) WS[i]+=WS[i-1];
    for(i=n-1;i>=0;i--) sa[--WS[x[i]]]=i;
    for(j=1,p=1;p<n;j*=2,m=p)
    {
        for(p=0,i=n-j;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
        for(i=0;i<n;i++) wv[i]=x[y[i]];
        for(i=0;i<m;i++) WS[i]=0;
        for(i=0;i<n;i++) WS[wv[i]]++;
        for(i=1;i<m;i++) WS[i]+=WS[i-1];
        for(i=n-1;i>=0;i--) sa[--WS[wv[i]]]=y[i];
        for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
        x[sa[i]] = cmp(y,sa[i-1],sa[i],j) ? p-1:p++;
    }
}
int rank[N],height[N];//height的有效值范围[2,n], rank的有效范围[0,n]
void calheight(int n)
{
    int i,j,k=0;
    for(i=1;i<=n;i++) rank[sa[i]]=i;
    for(i=0;i<n;height[rank[i++]]=k)
    for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
    return;
}
int a[N],dp[N][20];
void RMQ_init(int n)
{
    int i,j;
    for(i = 0; i < n; i++)  dp[i][0] = a[i];
    for(j = 1; (1<<j) <= n; j++)
        for(i = 0; i+(1<<j)-1 < n; i++)
            dp[i][j] = min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
}
int rmq(int l,int r)
{
    int k = 0;
    while( (1<<(k+1)) < r-l+1) k++;
    return min(dp[l][k], dp[r-(1<<k)+1][k]);
}
void calnum(int n,__int64 &ans)
{
    if(n == 0){
        ans++;
        return;
    }
    while(n)
    {
        ans++;
        n/=10;
    }
}
int main()
{
    while(scanf("%s",s) != EOF)
    {
        int n = strlen(s);
        int i;
        for(i = 0; i < n; i++)
            r[i] = s[i] - 'a' + 1;
        r[n] = 0;
        da(n+1,27);
        calheight(n);//注意不是n+1
       // for(i = 0; i < n; i++)
           // cout<<s+sa[i]<<endl;
        //cout<<"sa"<<endl;
        //for(i = 0; i <= n; i++)
          //  cout<<sa[i]<<" ";
        //cout<<endl;
        //cout<<"rank"<<endl;
        //for(i = 0; i <= n; i++)
          //  cout<<rank[i]<<" ";
        //cout<<endl;
        //cout<<"height"<<endl;
        //for(i = 0; i <= n; i++)
          //  cout<<height[i]<<" ";
        //cout<<endl;
        for(i = 1; i <= n; i++)
            a[i-1] = height[i];
        RMQ_init(n);
        int l1,r1,l2,r2;
        int t;
        scanf("%d",&t);
        __int64 ans1 = t,ans2 = 2*t+1;
        scanf("%d%d",&l1,&r1);
        ans1 += (r1-l1);
        ans2 += (r1-l1);
        t--;
        while(t--)
        {
            scanf("%d%d",&l2,&r2);
            ans1 += (r2 - l2);
            int temp;
            if(l2 == l1) temp = min(r1-l1,r2-l2);
            else {
                int k1 = rank[l1];
                int k2 = rank[l2];
                if(k2 < k1) swap(k1,k2);
                temp = rmq(k1,k2-1);
            }
            temp = min(temp, min(r1-l1,r2-l2) );
            calnum(temp,ans2);
            ans2 += (r2-l2-temp);
            l1 = l2;r1 = r2;
        }
        printf("%I64d %I64d\n",ans1,ans2);
    }
    return 0;
}