POJ3261Milk Patterns 后缀数组の变形

来源：互联网发布：linux系统有哪些编辑：程序博客网时间：2024/04/19 21:43

每日一题。。

这几天开始搞后缀数组，果然关键时候智商果然不够用，想到现在才有点眉目，虽然这种题千篇一律的模板，但是理解了才能运用的比较好，估计明天我就能自己写模板了。。

题意很明显，k个相同的子串，就是要在求height的时候多判断一下，过了。

#include<stdio.h>#include<string.h>#include<iostream>#define rep(i,n) for(int i = 0;i < n; i++)using namespace std;const int size  = 21000,INF = 1<<30,maxn=1000010;char s[size];int y;int rk[maxn],sa[maxn],height[maxn],w[size],wa[maxn],res[maxn];void getSa (int len,int up) {int *k = rk,*id = height,*r = res, *cnt = wa;rep(i,up) cnt[i] = 0;rep(i,len) cnt[k[i] = w[i]]++;rep(i,up) cnt[i+1] += cnt[i];for(int i = len - 1; i >= 0; i--) {sa[--cnt[k[i]]] = i;}int d = 1,p = 0;while(p < len){for(int i = len - d; i < len; i++) id[p++] = i;rep(i,len)if(sa[i] >= d) id[p++] = sa[i] - d;rep(i,len) r[i] = k[id[i]];rep(i,up) cnt[i] = 0;rep(i,len) cnt[r[i]]++;rep(i,up) cnt[i+1] += cnt[i];for(int i = len - 1; i >= 0; i--) {sa[--cnt[r[i]]] = id[i];}swap(k,r);p = 0;k[sa[0]] = p++;rep(i,len-1) {if(sa[i]+d < len && sa[i+1]+d <len &&r[sa[i]] == r[sa[i+1]]&& r[sa[i]+d] == r[sa[i+1]+d])k[sa[i+1]] = p - 1;else k[sa[i+1]] = p++;}if(p >= len) return ;d *= 2,up = p, p = 0;}}void getHeight(int len){    rep(i,len) rk[sa[i]] = i;    height[0] =  0;        int cn=1,flag=0,p=0,tmp;        for(int i = 0; i < len - 1; i++)        {        while(1)        {            flag=0;            if(i+p<len)            {                for(cn=1; cn<y; cn++)                {                 //   printf("?%d %d\n",w[i+p],w[tmp+p]);                 if(rk[i]>=cn)                 tmp=sa[rk[i]-cn];                 else {flag=1 ;break;}                    if(tmp+p<len)                    {                        if(w[i+p]!=w[tmp+p])                        {                            flag=1;                        break;                        }                    }                    else                    {                        flag=1;                        break;                    }                }                if(flag==1) break;                else p++;            }            else break;        }       // printf("%d",p);        height[rk[i]] = p;p = max(0,p - 1);//p=0;        }}int main(){    int n,up=0;    scanf("%d%d",&n,&y);    for(int i=0; i<n; i++)    {        scanf("%d",&w[i]);        if(w[i]>up) up=w[i];    }    w[n++]=0;    getSa(n,up+1);   // for(int i=0;i<n;i++)       //     printf("  %d   ",sa[i]);      //  printf("\n");    getHeight(n);    int ans=0;    for(int i=1; i<=n; i++)    {        ans=max(ans,height[i]);    }    printf("%d\n",ans);    return 0;}

0 0