poj 3261 Milk Patterns

最新推荐文章于 2019-04-10 21:03:43 发布

weixin_33965305

最新推荐文章于 2019-04-10 21:03:43 发布

阅读量53

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/sooflow/p/3382243.html

本文介绍了一种结合二分查找与后缀数组技术来解决问题的方法。通过构造后缀数组并利用高度数组进行判断，可以高效地解决特定类型的问题。代码示例展示了如何实现这些算法，并提供了一个具体的解决方案。

二分答案，利用height数组进行判定。

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 using namespace std;
 5 const int maxn=20000+10;
 6 const int maxvalue=1000000+10;
 7 int s[maxn];
 8 int sa[maxn],t[maxn],t2[maxn],c[maxvalue],n;
 9 int rank[maxn],height[maxn];
10 void build_sa(int m)
11 {
12     int i,*x=t,*y=t2;
13     for(i=0;i<m;i++) c[i]=0;
14     for(i=0;i<n;i++) c[x[i]=s[i]]++;
15     for(i=1;i<m;i++) c[i]+=c[i-1];
16     for(i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
17 
18     for(int k=1;k<=n;k<<=1)
19     {
20         int p=0;
21         for(i=n-k;i<n;i++) y[p++]=i;
22         for(i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
23 
24         for(i=0;i<m;i++) c[i]=0;
25         for(i=0;i<n;i++) c[x[y[i]]]++;
26         for(i=0;i<m;i++) c[i]+=c[i-1];
27         for(i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
28 
29         swap(x,y);
30         p=1;x[sa[0]]=0;
31         for(i=1;i<n;i++)
32             x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
33         if(p>=n) break;
34         m=p;
35     }
36 }
37 void getHeight()
38 {
39       int i,k=0;
40       for(i=0;i<n;i++) rank[sa[i]]=i;
41       for(i=0;i<n;i++)
42       {
43           if(k) k--;
44           int j=sa[rank[i]-1];
45           while(s[i+k]==s[j+k]) k++;
46           height[rank[i]]=k;
47       }
48 }
49 bool check(int k,int num)
50 {
51     int cnt=0;
52     for(int i=0;i<n;i++)
53     {
54         if(height[i]<k)
55             cnt=0;
56         else
57         {
58             cnt++;
59             if(cnt>=num-1) return true;
60         }
61     }
62     return false;
63 }
64 int main()
65 {
66     int k;
67     while(scanf("%d%d",&n,&k)!=EOF)
68     {
69         int maxv=0;
70         for(int i=0;i<n;i++)
71         {
72             scanf("%d",&s[i]);
73             if(s[i]>maxv) maxv=s[i];
74         }
75         build_sa(maxv+1);
76         getHeight();
77         int l=1,r=n/2;
78         int ans=-1;
79         while(l<=r)
80         {
81             int mid=(l+r)>>1;
82             if(check(mid,k))
83             {
84                 ans=mid;
85                 l=mid+1;
86             }
87             else
88                 r=mid-1;
89         }
90         printf("%d\n",ans);
91     }
92     return 0;
93 }