Milk Patterns POJ

Milk Patterns

POJ - 3261

第一道后缀数组，，有点蒙逼=_=

---

---

update:  2018-01-27

题意：求至少出现k次的最长子串的长度。

可以二分长度mid，看是否能出现k次。

如何判断出现k次？

由后缀数组的height数组， 将height数组按顺序分组，每一组内的最长公共前缀不小于mid（小于mid的自成一组），统计最大组的height个数即可。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define FP freopen("in.txt","r".stdin);

const int maxn=20010;
int s[maxn], s1[maxn];
int sa[maxn], t[maxn], t2[maxn], c[maxn], n, k;
int rank[maxn], height[maxn];

void build_sa(int m, int n){
    int i, *x=t, *y=t2;
    //基数排序
    for(i = 0; i < m; i++) c[i] = 0;
    for(i = 0; i < n; i++) c[x[i] = s[i]]++;
    for(i = 1; i < m; i++) c[i] += c[i-1];
    for(i = n-1;i >= 0; i--) sa[--c[x[i]]] = i;

    for(int k = 1; k <= n; k <<= 1){
        int p=0;
        //直接利用sa数组排序第二关键字
        for(i = n-k; i < n; i++) y[p++] = i;
        for(i = 0; i < n; i++) if(sa[i] >= k) y[p++] = sa[i] - k;

        //基数排序第一关键字
        for(i = 0; i < m; i++) c[i] = 0;
        for(i = 0; i < n; i++) c[x[y[i]]]++;
        for(i = 1; i < m; i++) c[i] += c[i-1];
        for(i = n-1; i >= 0; i--) sa[--c[x[y[i]]]] = y[i];
        //根据sa和y数组计算新的x数组
        swap(x, y);
        p = 1;
        x[sa[0]] = 0;
        for(i = 1; i < n; i++) 
          x[sa[i]] = y[sa[i-1]]==y[sa[i]] && y[sa[i-1]+k]==y[sa[i]+k] ? p-1 : p++;

        if(p >= n) break;  //以后即使继续倍增，sa也不会改变，推出
        m = p;          //下次基数排序的最大值
    }
}

void getHeight(int n) {
    int i, j, k = 0;
    for(i = 0; i <= n; i++) rank[sa[i]] = i;
    for(i = 0; i < n; i++) {
        if(k) k--;
        j = sa[rank[i]-1];
        while(s[i+k] == s[j+k]) k++;
        height[rank[i]] = k;
    }
}
int m; //模板长度

int main(){
    while(scanf("%d%d", &n, &k)!=EOF){
        for(int i = 0; i < n; i++) scanf("%d",&s[i]);
        for(int i = 0; i < n; i++) s1[i] = s[i];
        sort(s, s+n);
        int sz=unique(s,s+n)-s;
        for(int i = 0; i < n; i++) {
            s1[i]=lower_bound(s,s+sz,s1[i])-s+1;
        }
        for(int i = 0; i <= n; i++) s[i]=s1[i];
        build_sa(256, n+1);
        getHeight(n);
        int l=1, r=n;
        while(l <= r) { 
            int cnt=1;
            int m=(l+r)>>1;
            for(int i = 1; i <= n; i++){
                if(height[i] >= m) cnt++;
                else cnt=1;
                if(cnt>=k) break;
            }
            if(cnt >= k) l=m+1;
            else r=m-1;
        }
        printf("%d
", r);
    }

}

也可用哈希做，慢

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ull unsigned long long
const int maxn=20010;
const int seed=131;

ull base[maxn],h[maxn];
int a[maxn];
int n,k;

int check(int m)
{
    memset(h,0,sizeof(h));
    for(int i=0;i<m;i++) h[m-1]=h[m-1]*seed+a[i];
    for(int i=m;i<n;i++){
        h[i]=h[i-1]*seed-a[i-m]*base[m]+a[i];
    }
    sort(h,h+n);
    int cnt=1;
    for(int i=n-1;i>=m;i--){
        if(h[i]==h[i-1]) cnt++;
        else cnt=1;
        if(cnt>=k) return 1;
    }
    return 0;
}
int main()
{
    base[0]=1;
    for(int i=1;i<maxn;i++) base[i]=base[i-1]*seed;
    while(scanf("%d%d",&n,&k)!=EOF){
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
        }
        int L=0,R=n;
        while(L<=R){
            int m=L+(R-L)/2;
            if(check(m)) L=m+1;
            else R=m-1;
        }
        printf("%d
",R);
    }
}