bzoj 1717 [USACO06DEC] Milk Patterns (后缀数组+二分)

题目大意:求可重叠的相同子串数量至少是K的子串最长长度  

洛谷传送门

依然是后缀数组+二分,先用后缀数组处理出height

每次二分出一个长度x,然后去验证,在排序的后缀串集合里,有没有连续数量多于K个串的长度>=x,

但据说有一种高端做法是把二分换成单调队列,能减少常数,可惜我并没有看懂......

原题好像是哈希的骚操作,但网上的题解好像都是后缀数组......

比上一道男人八题简单多了,我原来的错代码竟然卡过去了70分..

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 #define N 20010
 5 #define inf 0x3f3f3f3f
 6 #define maxn 1000010
 7 using namespace std;
 8 
 9 int n,len,tot,K;
10 int a[N],tr[N],sa[N],rk[N],hs[N],h[N];
11 struct node{int id,val,w;}d[N];
12 int cmp1(node s1,node s2){return s1.val<s2.val;}
13 int cmp2(node s1,node s2){return s1.id<s2.id;}
14 int gint()
15 {
16     int rett=0,fh=1;char c=getchar();
17     while(c<'0'||c>'9'){if(c=='-')fh=-1;c=getchar();}
18     while(c>='0'&&c<='9'){rett=(rett<<3)+(rett<<1)+c-'0';c=getchar();}
19     return rett*fh;
20 }
21 bool check(int k,int x,int y){
22     if(x+k>len||y+k>len) return 0;
23     else return (rk[x]==rk[y]&&rk[x+k]==rk[y+k])?1:0;
24 }
25 void get_sa()
26 {
27     int cnt=0,i;
28     for(i=1;i<=len;i++) hs[a[i]]++;
29     for(i=1;i<=tot;i++) if(hs[i]) tr[i]=++cnt;
30     for(i=1;i<=tot;i++) hs[i]+=hs[i-1];
31     for(i=1;i<=len;i++) rk[i]=tr[a[i]],sa[hs[a[i]]--]=i;
32     for(int k=1;cnt<len;k<<=1)
33     {
34         for(i=1;i<=cnt;i++) hs[i]=0;
35         for(i=1;i<=len;i++) hs[rk[i]]++;
36         for(i=1;i<=cnt;i++) hs[i]+=hs[i-1];
37         for(i=len;i>=1;i--) if(sa[i]>k) tr[sa[i]-k]=hs[rk[sa[i]-k]]--;
38         for(i=1;i<=k;i++) tr[len-i+1]=hs[rk[len-i+1]]--;
39         for(i=1;i<=len;i++) sa[tr[i]]=i;
40         for(i=1,cnt=0;i<=len;i++) tr[sa[i]]=check(k,sa[i],sa[i-1])?cnt:++cnt;
41         for(i=1;i<=len;i++) rk[i]=tr[i];
42     }
43 }
44 void get_height()
45 { 
46     for(int i=1;i<=n;i++){
47         if(rk[i]==1) continue;
48         for(int j=max(1,h[rk[i-1]]-1);;j++)
49             if(a[i+j-1]==a[sa[rk[i]-1]+j-1]) h[rk[i]]=j;
50             else break;
51     }
52 }
53 void descrete()
54 {
55     sort(d+1,d+n+1,cmp1);
56     d[0].val=-1;
57     for(int i=1;i<=n;i++)
58         if(d[i].val==d[i-1].val)
59             d[i].w=d[i-1].w;
60         else d[i].w=++tot;
61     sort(d+1,d+n+1,cmp2);
62     for(int i=1;i<=n;i++)
63         a[i]=d[i].w;
64 }
65 int check(int ans)
66 {
67     int i,cnt=0;
68     for(i=1;i<=n;)
69     {
70         if(h[i]<ans) {i++;continue;}
71         for(cnt=0;h[i]>=ans&&i<=n;i++) 
72             cnt++;
73         if(cnt>=K-1) return 1;
74     }return 0;
75 }
76 
77 int main()
78 {
79     scanf("%d%d",&n,&K);len=n;
80     for(int i=1;i<=n;i++) 
81         d[i].val=gint(),d[i].id=i;
82     descrete();
83     get_sa();
84     get_height();
85     int l=0,r=n,ans=0;
86     while(l<=r){
87         int mid=(l+r)>>1;
88         if(check(mid)) ans=mid,l=mid+1;
89         else r=mid-1;
90     }printf("%d\n",ans);
91     return 0;
92 }

猜你喜欢

转载自www.cnblogs.com/guapisolo/p/9697166.html