第三场1010题

http://acm.hdu.edu.cn/showproblem.php?pid=4622

#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
#define sz 2010
int tmp[sz],top[sz];
int sa[sz],r[sz],h[sz];
void get_suffix(char s[],int sa[],int r[],int h[],int len)
{
    int n=len;
    int i,j,p,na=n<256?256:n;
    memset(top,0,na*sizeof(int));
    for(i=0;i<n;i++)top[r[i]=s[i]&0xff]++;
    for(i=0;i<na;i++)top[i]+=top[i-1];
    for(i=0;i<n;i++)sa[--top[r[i]]]=i;
    for(p=1;p<n;p<<=1)
    {
        for(i=0;i<n;i++)
        {
            j=sa[i]-p;
            if(j<0)j+=n;
            tmp[top[r[j]]++]=j;
        }
        sa[tmp[top[0]=0]]=j=0;
        for(i=1;i<n;i++)
        {
            if(r[tmp[i]]!=r[tmp[i-1]]||r[tmp[i]+p]!=r[tmp[i-1]+p])
                top[++j]=i;
            sa[tmp[i]]=j;
        }
        memcpy(r,sa,n*sizeof(int));
        memcpy(sa,tmp,n*sizeof(int));
        if(j>=n-1)break;
    }
    //height=lcp(sa[i-1],sa[i]);
    for(j=r[h[i=p=0]=0];i<n-1;i++,p++)
    {
        while(p>=0&&s[i]!=s[sa[j-1]+p])
        {
            h[j]=p--;
            j=r[sa[j]+1];
        }
    }
}
int ans[sz][sz];
int tnt[sz][sz],nt[sz][sz];
char str[sz];
#define lowbit(a) ((a)&-(a))
void add(int st,int pos,int val)
{
    while(pos<sz)
    {
        nt[st][pos]+=val;
        pos+=lowbit(pos);
    }
}
int query(int st,int pos)
{
    int ret=0;
    while(pos>0)
    {
        ret+=nt[st][pos];
        pos-=lowbit(pos);
    }
    return ret;
}
void insert(int st,int pos,int len)
{
    add(st,pos,1);
    add(st,pos+len,-1);
}
int main()
{
    int ti;scanf("%d",&ti);
    for(int ca=1;ca<=ti;ca++)
    {
        memset(nt,0,sizeof(nt));
        memset(tnt,0,sizeof(tnt));
        memset(ans,0,sizeof(ans));
        scanf("%s",str);
        int n=strlen(str);
        get_suffix(str,sa,r,h,n+1);
        for(int j=0;j<n;j++)
        {
            int H=0x3f3f3f3f,pre=0;
            for(int k=r[j]-1;k>0;k--)
            {
                H=min(H,h[k+1]);
                if(h==0||pre==j)break;
                if(sa[k]<j)
                {
                    for(int i=pre;i<=sa[k];i++)
                    {
                        tnt[i][j]=max(tnt[i][j],H);
                    }
                    pre=max(pre,sa[k]+1);
                }
            }
            H=0x3f3f3f3f,pre=0;
            for(int k=r[j]+1;k<=n;k++)
            {
                H=min(H,h[k]);
                if(h==0||pre==j)break;
                if(sa[k]<j)
                {
                    for(int i=pre;i<=sa[k];i++)
                    {
                        tnt[i][j]=max(tnt[i][j],H);
                    }
                    pre=max(pre,sa[k]+1);
                }
            }
        }
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                insert(i+1,j+1,tnt[i][j]);
                //printf("%d ",tnt[i][j]);
            }
            //printf("\n");
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                ans[i][j]=ans[i][j-1]+query(i,j);
                //printf("%d ",ans[i][j]);
            }
            //printf("\n");
        }
        int q;scanf("%d",&q);
        for(int i=0;i<q;i++)
        {
            int l,r;scanf("%d%d",&l,&r);
            int zz=r-l+1;
            zz=zz*(zz+1)/2;
            printf("%d\n",zz-ans[l][r]);
        }
    }
}

比赛时未考虑相隔位置的计数

#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
#define sz 2010
int tmp[sz],top[sz];
int sa[sz],r[sz],h[sz];
void get_suffix(char s[],int sa[],int r[],int h[],int len)
{
    int n=len;
    int i,j,p,na=n<256?256:n;
    memset(top,0,na*sizeof(int));
    for(i=0;i<n;i++)top[r[i]=s[i]&0xff]++;
    for(i=0;i<na;i++)top[i]+=top[i-1];
    for(i=0;i<n;i++)sa[--top[r[i]]]=i;
    for(p=1;p<n;p<<=1)
    {
        for(i=0;i<n;i++)
        {
            j=sa[i]-p;
            if(j<0)j+=n;
            tmp[top[r[j]]++]=j;
        }
        sa[tmp[top[0]=0]]=j=0;
        for(i=1;i<n;i++)
        {
            if(r[tmp[i]]!=r[tmp[i-1]]||r[tmp[i]+p]!=r[tmp[i-1]+p])
                top[++j]=i;
            sa[tmp[i]]=j;
        }
        memcpy(r,sa,n*sizeof(int));
        memcpy(sa,tmp,n*sizeof(int));
        if(j>=n-1)break;
    }
    //height=lcp(sa[i-1],sa[i]);
    for(j=r[h[i=p=0]=0];i<n-1;i++,p++)
    {
        while(p>=0&&s[i]!=s[sa[j-1]+p])
        {
            h[j]=p--;
            j=r[sa[j]+1];
        }
    }
}
int ans[sz][sz];
int tnt[sz][sz],nt[sz][sz];
char str[sz];
#define lowbit(a) ((a)&-(a))
void add(int st,int pos,int val)
{
    while(pos<sz)
    {
        nt[st][pos]+=val;
        pos+=lowbit(pos);
    }
}
int query(int st,int pos)
{
    int ret=0;
    while(pos>0)
    {
        ret+=nt[st][pos];
        pos-=lowbit(pos);
    }
    return ret;
}
void insert(int st,int pos,int len)
{
    add(st,pos,1);
    add(st,pos+len,-1);
}
int main()
{
    //freopen("1002.in","r",stdin);
    //freopen("wrong","w",stdout);
    int ti;scanf("%d",&ti);
    for(int ca=1;ca<=ti;ca++)
    {
        memset(nt,0,sizeof(nt));
        memset(tnt,0,sizeof(tnt));
        memset(ans,0,sizeof(ans));
        scanf("%s",str);
        int n=strlen(str);
        get_suffix(str,sa,r,h,n+1);
        int q;scanf("%d",&q);
        while(q--)
        {
            int l,r;
            scanf("%d%d", &l, &r);
            l--, r--;
            int pre = -1, qlen = r - l + 1;
            int ans = (qlen + 1) * qlen / 2;
            int H=0x3f3f3f3f;
            for(int i = 1; i <= n; i++)
            {
                H=min(H,h[i]);
                if(sa[i] >= l && sa[i] <= r)
                {
                    if(pre != -1)
                    {
                        int tmp=H;
                        tmp=min(tmp, r - pre + 1);
                        ans-=min(tmp, r - sa[i] + 1);
                        if(tmp<r-sa[i]+1)
                        {
                            pre=sa[i];
                            H=0x3f3f3f3f;
                        }
                    }
                    else
                    {
                        pre=sa[i];
                        H=0x3f3f3f3f;
                    }
                }
            }
            printf("%d\n", ans);
        }

    }
}


/*
2
htopotmitnqwtoeton
1
1 16
htopotmitnqwtoeto
1
1 16

2
yyzrompymegsys
1
1 11
yyzrompymegsy
1
1 11
*/

另一种做法 和一些数据