[Usaco2006 Dec]Milk Patterns

[Usaco2006 Dec]Milk Patterns

Description

农夫John发现他的奶牛产奶的质量一直在变动。经过细致的调查，他发现：虽然他不能预见明天 
产奶的质量，但连续的若干天的质量有很多重叠。我们称之为一个“模式”。 
John的牛奶按质量可以被赋予一个0到1000000之间的数。并且John记录了N(1<=N<=20000)天的 
牛奶质量值。他想知道最长的出现了至少K(2<=K<=N)次的模式的长度。 
比如1 2 3 2 3 2 3 1 中 2 3 2 3出现了两次。当K=2时，这个长度为4。

Input

* Line 1: 两个整数 N,K。 
* Lines 2..N+1: 每行一个整数表示当天的质量值。

Output

* Line 1: 一个整数：N天中最长的出现了至少K次的模式的长度

Sample Input

8 2
1
2
3
2
3
2
3
1

Sample Output

4

Source

后缀数组

答案应该是height数组中连续K-1个值的最小值的最大值。

这题还是要先离散化一下，不然可能会MLE

code：

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #define maxn 30010
 using namespace std;
 char ch;
 int n,k,max_val,tot,head,tail,s[maxn],num[maxn],SA[maxn],rank[maxn],tmp[maxn],sum[maxn],height[maxn];
 struct DATA{
     int val,tim;
 }list[maxn];
 inline bool isdigit(char ch){return ''<=ch&&ch<='';}
 inline void read(int &x){
     for (ch=getchar();!isdigit(ch);ch=getchar());
     for (x=;isdigit(ch);x=x*+ch-'',ch=getchar());
 }
 inline bool cmp(int a,int b){return s[a]<s[b];}
 void prepare(){
     sort(num+,num++n,cmp);
     for (int i=,pre=-;i<=n;i++){
         if (s[num[i]]!=pre) pre=s[num[i]],++max_val;
         s[num[i]]=max_val;
     }
 }
 void get_SA(){
     for (int i=;i<=n;i++) sum[tmp[i]=s[i]]++;
     for (int i=;i<=max_val;i++) sum[i]+=sum[i-];
     for (int i=n;i>=;i--) SA[sum[tmp[i]]--]=i;
     rank[SA[]]=tot=;
     for (int i=;i<=n;i++){
         if (tmp[SA[i]]!=tmp[SA[i-]]) tot++;
         rank[SA[i]]=tot;
     }
     for (int len=;len<=n;len<<=){
         memset(sum,,sizeof(sum));
         for (int i=;i<=n;i++) sum[rank[i+len]]++;
         for (int i=;i<=n;i++) sum[i]+=sum[i-];
         for (int i=n;i>=;i--) tmp[sum[rank[i+len]]--]=i;
         memset(sum,,sizeof(sum));
         for (int i=;i<=n;i++) sum[rank[tmp[i]]]++;
         for (int i=;i<=n;i++) sum[i]+=sum[i-];
         for (int i=n;i>=;i--) SA[sum[rank[tmp[i]]]--]=tmp[i];
         tmp[SA[]]=tot=;
         for (int i=;i<=n;i++){
             if (rank[SA[i]]!=rank[SA[i-]]||rank[SA[i]+len]!=rank[SA[i-]+len]) tot++;
             tmp[SA[i]]=tot;
         }
         for (int i=;i<=n;i++) rank[i]=tmp[i];
     }
 }
 void get_height(){
     for (int i=,j=;i<=n;i++){
         if (rank[i]==) continue;
         while (s[i+j]==s[SA[rank[i]-]+j]) j++;
         height[rank[i]]=j;
         if (j>) j--;
     }
 }
 int main(){
     read(n),read(k);
     for (int i=;i<=n;i++) read(s[i]),num[i]=i;
     prepare(),get_SA(),get_height();
     k--,head=,tail=,max_val=;
     for (int i=;i<=n;i++){
         while (head<=tail&&i-list[head].tim>=k) head++;
         while (head<=tail&&height[i]<list[tail].val) tail--;
         list[++tail]=(DATA){height[i],i};
         if (i>k&&max_val<list[head].val) max_val=list[head].val;
     }
     printf("%d\n",max_val);
     return ;
 }