POJ3261 Milk Patterns
阿新 • • 發佈：2019-03-05
如果 algorithm node bool amp 否則 += mem [1]
思路
後綴數組+二分答案的裸題
二分長度mid，然後如果至少有k個連續的height>=mid，就是true
否則就是false
代碼
#include <cstdio> #include <algorithm> #include <cstring> using namespace std; struct Node{ int pos,r[2]; }midx[100000],x[100000]; int barrel[100000],sa[100000],ranks[100000],height[100000],n,k,s[100000]; int c_sort(int n,int lim){ for(int i=0;i<2;i++){ memset(barrel,0,sizeof(barrel)); for(int j=1;j<=n;j++) barrel[x[j].r[i]]++; for(int j=1;j<=lim;j++) barrel[j]+=barrel[j-1]; for(int j=n;j>=1;j--) midx[barrel[x[j].r[i]]--]=x[j]; for(int j=1;j<=n;j++) x[j]=midx[j]; } ranks[x[1].pos]=1; int cnt=1; for(int i=2;i<=n;i++){ if(x[i].r[0]==x[i-1].r[0]&&x[i].r[1]==x[i-1].r[1]) ranks[x[i].pos]=cnt; else ranks[x[i].pos]=++cnt; } return cnt; } void get_sa(int n){ for(int i=1;i<=n;i++){ x[i]=(Node){i,0,s[i]}; } int cnt=c_sort(n,255); for(int i=1;cnt<n;i<<=1){ for(int j=1;j<=n;j++) x[j]=(Node){j,(i+j<=n)?ranks[i+j]:0,ranks[j]}; cnt=c_sort(n,cnt); } for(int i=1;i<=n;i++) sa[ranks[i]]=i; for(int i=1,j=0,k;i<=n;height[ranks[i++]]=j) for((j)?j--:0,k=sa[ranks[i]-1];s[i+j]==s[j+k];j++); } bool check(int x){ int t=1,ans=1; for(int i=2;i<=n;i++){ if(height[i]>=x){ t++; if(t>=k) return true; } else{ t=1; } } // printf("x=%d ans=%d\n",x,ans); return false; } int main(){ scanf("%d %d",&n,&k); for(int i=1;i<=n;i++){ scanf("%d",&s[i]); } get_sa(n); int ans=0,l=0,r=n; while(l<=r){ int mid=(l+r)>>1; if(check(mid)) ans=mid,l=mid+1; else r=mid-1; } printf("%d\n",ans); return 0; }
POJ3261 Milk Patterns