hdu 6194 - 后缀数组+线段树

题目链接:点击打开链接

题解思路:用后缀数组处理得到的hight数组后,对它再进行线段建树，维护的是区间内最小的一个值，那么我们每次取枚举一个长度为n的区间时，取它的最小值这表示从1-ans_min的字符串长度都是满足的，那么题目还要求正好n次重复，那么我们就再取两次一个的左边界加一的区间值，一个是右边界加一的区间值，取两值的max那么此区间可取范围就应该是(左右边界+1的max,ans_min],时间复杂度就应该为lenlog(len)

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int mx = 1e5+10;
char str[mx];
int c[mx],wa[mx],wb[mx],sa[mx],hight[mx],ran[mx];
void sort_sa(int m,int n){
    int *x=wa,*y=wb,p=0;
    for(int i=0;i<m;i++)  c[i]=0;
    for(int i=0;i<n;i++)  c[x[i]=str[i]]++;
    for(int i=1;i<m;i++)  c[i]+=c[i-1];
    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    for(int k=1;p<n;k<<=1,m=p){
        p=0;
        for(int i=n-k;i<n;i++) y[p++]=i;//按个位排序
        for(int i=0;i<n;i++)  if(sa[i]>=k) y[p++]=sa[i]-k;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[y[i]]]++;
        for(int i=1;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--)  sa[--c[x[y[i]]]]=y[i];
        swap(x,y);x[sa[0]] = 0,p = 1;
        for(int i=1;i<n;i++)
        if(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]) x[sa[i]]=p-1;
        else x[sa[i]]=p++;
    }
}
void Gethight(){
    int k=0,len=strlen(str);
    for(int i=1;i<=len;i++) ran[sa[i]]=i;
    for(int i=0;i<len;i++){
        if(k) k--;
        int j=sa[ran[i]-1];
        while(str[j+k]==str[i+k]) k++;
        hight[ran[i]]=k;
    }
}
int sum[mx<<2];
void build(int l,int r,int rt){
    if(l==r){
        sum[rt] = hight[l];
        return ;
    }
    int mid = (l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    sum[rt] = min(sum[rt<<1],sum[rt<<1|1]);
}
int query(int L,int R,int l,int r,int rt){
    if(L<=l&&R>=r) return sum[rt];
    int mid = (l+r)>>1;
    int ans = 1e9;
    if(mid>=L) ans = query(L,R,l,mid,rt<<1);
    if(mid<R) ans = min(ans,query(L,R,mid+1,r,rt<<1|1));
    return ans;
}
int main(){
    int t,n;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        scanf("%s",str);
        int len=strlen(str),ans = 0;
        sort_sa(200,len+1);
        Gethight();
        hight[len+1] = 1e9;
        build(1,len,1);
        if(n>len) { puts("0");  continue;  }
        if(n==1){
            ans += len - sa[1] - hight[2];
            ans += len - sa[len] - hight[len];
            for(int i=2;i<len;i++)
            ans += len - sa[i] - max(hight[i],hight[i+1]);
            printf("%d\n",ans);
            continue;
        }
        int R = query(2,n,1,len,1),L = min(R,hight[n+1]);
        ans += R-L;
        for(int i=3;i<=len-n+1;i++){
            R = query(i,i+n-2,1,len,1),L = max(min(R,hight[i+n-1]),min(R,hight[i-1]));
            ans += R-L;
        }
        R = query(len-n+2,len,1,len,1),L = min(R,hight[len-n+1]);
        ans += R-L;
        printf("%d\n",ans);
    }
    return 0;
}