POJ 3261 字符串上的k次覆盖问题

题目大意：

给定一个数组，求一个最大的长度的子串至少出现过k次

 

一个子串出现多次，也就是说必然存在2个子串间的前缀长度为所求的值

通过二分答案，通过线性扫一遍，去判断出现次数，也就是说每次遇见一个height[i] , 出现次数就加1，否则重置为1

 

 1 #include <cstdio>

 2 #include <cstring>

 3 #include <iostream>

 4 using namespace std;

 5 const int N = 20010;

 6 int rank[N] , sa[N] , height[N];

 7 int wa[N] , wb[N] , wsf[N] , wv[N];

 8 int a[N];

 9 

10 int cmp(int *r , int a , int b , int l)

11 {

12     return r[a]==r[b] && r[a+l]==r[b+l];

13 }

14 

15 void getSa(int *r , int *sa , int n , int m)

16 {

17     int i,j,p;

18     int *x=wa , *y=wb , *t;

19     for(i=0 ; i<m ; i++) wsf[i]=0;

20     for(i=0 ; i<n ; i++) wsf[x[i]=r[i]]++;

21     for(i=1 ; i<m ; i++) wsf[i] += wsf[i-1];

22     for(i=n-1 ; i>=0 ; i--) sa[--wsf[x[i]]]=i;

23 

24     p=1;

25     for(j=1 ; p<n ; j*=2 , m=p){

26         for(p=0 , i=n-j ; i<n ; i++) y[p++]=i;

27         for(i=0 ; i<n ; i++) if(sa[i]>=j) y[p++]=sa[i]-j;

28 

29         for(i=0 ; i<n ; i++) wv[i]=x[y[i]];

30         for(i=0 ; i<m ; i++) wsf[i]=0;

31         for(i=0 ; i<n ; i++) wsf[wv[i]]++;

32         for(i=1 ; i<m ; i++) wsf[i]+=wsf[i-1];

33         for(i=n-1 ; i>=0 ; i--) sa[--wsf[wv[i]]]=y[i];

34 

35         t=x,x=y,y=t;

36         x[sa[0]]=0;

37         for(p=1 , i=1 ; i<n ; i++)

38             x[sa[i]] = cmp(y , sa[i-1] , sa[i] , j)?p-1:p++;

39     }

40     return ;

41 }

42 

43 void getHeight(int *r , int *sa , int n)

44 {

45     for(int i=1 ; i<=n ; i++) rank[sa[i]]=i;

46     int k=0;

47     int j;

48     for(int i=0 ; i<n ; height[rank[i++]]=k)

49         for(k?k--:0 , j=sa[rank[i]-1] ; r[i+k]==r[j+k] ; k++);

50     return;

51 }

52 

53 bool check(int m , int n , int k)

54 {

55     int cnt=1;

56     for(int i=1 ; i<=n ; i++){

57         if(height[i]>=m){

58             cnt++;

59          //   cout<<"here: "<<m<<" "<<sa[i]<<" "<<sa[i-1]<<endl;

60             if(cnt>=k) return true;

61         }

62         else {cnt=1;if(cnt>=k) return true;}

63     }

64     return false;

65 }

66 

67 int main()

68 {

69   //  freopen("a.in" , "r" , stdin);

70     int n,k;

71     while(~scanf("%d%d" , &n , &k))

72     {

73         int maxn =0 ;

74         for(int i=0 ; i<n ; i++){

75             scanf("%d" , a+i);

76             a[i]++;

77             maxn = max(maxn , a[i]);

78         }

79         a[n]=0;

80         getSa(a , sa , n+1 , maxn+1);

81         getHeight(a , sa , n);

82 

83         int l=0 , r=n , ans=0;

84         while(l <= r){

85             int m=(l+r)>>1;

86             if(check(m,n,k)){

87                 ans = m;

88                 l=m+1;

89             }else r=m-1;

90         }

91         printf("%d\n" , ans);

92     }

93     return 0;

94 }