HDU 4691 Front compression(后缀数组 LCA)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4691

这个题目题目看错只能是TLE 或者MLE

看错题目就是前面所有的插入的前缀，那么用后缀数组TLE

用trie MLE，反正题目理解错了怎么也过不了

其实正解就是和上面一个输入

两种方法：后缀数组 结合RMQ 暴力水过

用RMQ的时候注意一下，在初始化的时候n是要包括的

在查询的时候l++，因为height是当前的和前面的最长前缀部分

后缀数组：

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <math.h>
#include <string.h>
using namespace std;
#define maxn 110000
#define ws ws1
int wa[maxn],wb[maxn],wv[maxn],ws[maxn];
int cmp(int *r,int a,int b,int l)
{return r[a]==r[b]&&r[a+l]==r[b+l];}
void da(char *r,int *sa,int n,int m)
{
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++;
}
return;
}
int rank[maxn],height[maxn];
void calheight(char *r,int *sa,int n)
{
int i,j,k=0;
for(i=1;i<=n;i++) rank[sa[i]]=i;//在计算height的时候顺便就把rank计算出来了，反正也要用
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;
}
char rec[maxn];
int sa[maxn];
int n;
int rmq[maxn][20];
int RMQ(int n)
{
int i,j;
for(i=0;i<=n;i++)
rmq[i][0]=i;
for(j=1;(1<<j)<=n;j++)
for(i=0;i+(1<<j)-1<=n;i++)
{
if(height[rmq[i][j-1]]<=height[rmq[i+(1<<(j-1))][j-1]])
rmq[i][j]=rmq[i][j-1];
else
rmq[i][j]=rmq[i+(1<<(j-1))][j-1];
}
return 0;
}
int query(int l,int r)
{
    if(l>r)
    l^=r,r^=l,l^=r;
    if(l != r) l++;
int k=log(double(r-l+1))/log(2.0);
if(height[rmq[l][k]] <= height[rmq[r-(1<<k)+1][k]])
return rmq[l][k];
else
return rmq[r-(1<<k)+1][k];
}
int find_ans(int k)
{
    if(k<10)
    return 1;
    int ans=0;
    while(k)
    {
        ans++;
        k/=10;
    }
    return ans;
}
int main()
{
    int i,j,k,a,b,c,d,num;
    int mid,m;
    long long ans1,ans2;
    while(scanf("%s",rec)!=EOF)
    {
        n=strlen(rec);
        ans1=ans2=0;
        da(rec,sa,n+1,200);//这里是n+1 因为看这个函数里面是 < n 的
        calheight(rec,sa,n);//注意这里面是 n 了因为看函数里面是 <=n 的，所以这里要注意
        //for(i=0;i<n;i++)
        //printf("%d ",height[i]);
        //printf("\n");
        RMQ(n);
        scanf("%d",&m);
        scanf("%d%d",&a,&b);
        c=a,d=b;
        ans1=b-a+1;ans2=b-a+2+1;
        for(i=1;i<m;i++)
        {
            scanf("%d%d",&a,&b);
            if(rank[a]!=rank[c])
            {
            num=query(rank[a],rank[c]);
            num=height[num];
            if(num>b-a)
            num=b-a;
            if(num>d-c)
            num=d-c;
            }
            else
            {
                num=min(b-a,d-c);
            }
            ans1+=b-a+1;
            ans2+=(b-a-num+2)+find_ans(num);
            c=a,d=b;
           // printf("ans:%d\n",num);
        }
        printf("%I64d %I64d\n",ans1,ans2);
    }
    return 0;
}

暴力：

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <cmath>
#include <algorithm>
using namespace std;
#define maxn 510000
char str[maxn];
int find_ans(int k)
{
    if(k<10)
    return 1;
    int ans=0;
    while(k)
    {
        ans++;
        k/=10;
    }
    return ans;
}
int find_max(int a,int b,int c,int d)
{
    int ans=0;
    int i,j;
    for(i=a,j=c;i<b && j<d;i++,j++)
    if(str[i]==str[j])
    ans++;
    else
    return ans;
    return ans;
}
int main()
{
    long long ans1,ans2;
    int n,a,b,len1,len2,c,d,num,i,j,k;
    while(scanf("%s",str)!=EOF)
    {
        ans1=ans2=0;
        scanf("%d",&n);
        c=0,d=0;
        for(i=0;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            ans1+=b-a+1;
            if(i==0)
            {
                ans2+=b-a+2+1;
            }
            else if(c==a)
            {
                num=min(b-a,d-c);// printf("ans:%d %d\n",num,find_ans(num));
                ans2+=(b-a-num+2)+find_ans(num);
            }
            else
            {
                num=find_max(a,b,c,d);
                ans2+=(b-a-num+2)+find_ans(num);
            }
            c=a,d=b;
        }
        prinf("%I64d %I64d\n",ans1,ans2);
        //cout<<ans1<<' '<<ans2<<endl;
    }
    return 0;
}