bzoj1717(后缀数组)

1717: [Usaco2006 Dec]Milk Patterns 产奶的模式

Time Limit: 5 Sec   Memory Limit: 64 MB
Submit: 856   Solved: 470
[ Submit][ Status][ Discuss]

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

解题思路：先后缀数组求出height数组，然后二分答案，用单调队列判断。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
int n,m,k;
int height[20001],sa[20001],rank[20001],zhi[20001];
int q[1000000],zan[20001];


inline int read()
{
char y; int x=0,f=1; y=getchar();
while (y<'0' || y>'9') {if (y=='-') f=-1; y=getchar();}
while (y>='0' && y<='9') {x=x*10+int(y)-48; y=getchar();}
return x*f;
}


bool cmp(int x,int y)
 {
  if (rank[x]<rank[y]) return true;else
  if (rank[x]==rank[y])
   {
    int u1,u2;
      if (x+k<=n) u1=rank[x+k];else u1=-1;
      if (y+k<=n) u2=rank[y+k];else u2=-1;
      return u1<u2;
 }
return false;
 }


void makeheight()
 {
  int j=0;
  for (int i=1;i<=n;++i)
  {
  if (j) --j;
  int u1=sa[rank[i]+1];
  if (u1==0) continue;
  while(zhi[i+j]==zhi[u1+j] && i+j<=n && u1+j<=n) ++j;
  height[rank[i]]=j;
}
 }


int main()
{
   n=read(); m=read();
   for (int i=1;i<=n;++i)
    {
     int x=read();
     rank[i]=x; sa[i]=i; zhi[i]=x;
}
   for (k=1;k*2<=n;k=k*2)
    {
    sort(sa+1,sa+n+1,cmp); zan[sa[1]]=1;
    for (int i=2;i<=n;++i)
    {
      if (cmp(sa[i-1],sa[i])) zan[sa[i]]=zan[sa[i-1]]+1;else
       zan[sa[i]]=zan[sa[i-1]];
}
for (int i=1;i<=n;++i) rank[i]=zan[i];
}
   makeheight(); 
   int l=0; int r=n; 
   while (r>l)
    {
    int mid=(l+r)/2;
    int tail=0,head=0;
    bool mg=false; 
    for (int i=1;i<=n-1;++i)
    {
    while (head<tail && q[head]<i)
    {
    ++head;
 }
    if (tail==head)
    {
      if (height[i]>=mid) ++tail,q[tail]=i;
}
   if (tail!=head)
        {
        while (height[q[tail]+1]>=mid) ++tail,q[tail]=q[tail-1]+1;
}
if (tail-head+1>=m) 
{
mg=true; break;
}
}
if (mg) l=mid+1;else r=mid;
}
   printf("%d",l-1);
}