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;}