P2852 後綴數組 + 單調隊列

題目傳送門

題意:

 \dpi{150}n 個數的序列 \dpi{150}x ,問你出現次數至少爲 \dpi{150} k 的子序列的最大長度。

數據範圍: \dpi{150}1 \leqslant n \leqslant 2 \cdot 10^4 \;,\; 0 \leqslant k \leqslant 10^6 \;,\; 2 \leqslant k \leqslant n 。

題解:

首先貪心的思考一下,把題意中的“至少”二字去掉。

這個相當於問 \dpi{150} min\{height[j]\} \; j \in [i , i + k - 2] 的最大值。

那我們掏出後綴數組的板子,然後求出 \dpi{150} height 數組。

然後我們用單調隊列維護一下區間最小值的最大值就好了。

感受:

看了之後立刻想到了和題解一樣的解法。

但是說實話我第一反應不是實現它,還是hack我自己。

不自信。

代碼:

#include<bits/stdc++.h>
using namespace std ;
typedef long long ll ;
const int maxn = 1e6 + 5 ;
int rk[maxn << 1] , sa[maxn << 1] , height[maxn << 1] ;
int tmp[maxn << 1] , cnt[maxn] ;
int s[maxn] ;
int q[maxn] ;
void suffixarray(int n , int m)
{
   n ++ ;
   for(int i = 0 ; i < n * 2 + 5 ; i ++)
     rk[i] = sa[i] = height[i] = tmp[i] = 0 ;//開2 倍空間
   for(int i = 0 ; i < m ; i ++)  cnt[i] = 0 ;
   for(int i = 0 ; i < n ; i ++)  cnt[rk[i] = s[i]] ++ ;
   for(int i = 1 ; i < m ; i ++)  cnt[i] += cnt[i - 1] ;
   for(int i = 0 ; i < n ; i ++)  sa[-- cnt[rk[i]]] = i ;
   for(int k = 1 ; k <= n ; k <<= 1)
   {
     int j = 0 ;
     for(int i = 0 ; i < n ; i ++)
     {
       j = sa[i] - k ;
       if(j < 0)  j += n ;
       tmp[cnt[rk[j]] ++] = j ;
     }
     sa[tmp[cnt[0] = 0]] = j = 0 ;
     for(int i = 1 ; i < n ; i ++)
     {
       if(rk[tmp[i]] != rk[tmp[i - 1]]
       || rk[tmp[i] + k] != rk[tmp[i - 1] + k])
         cnt[++ j] = i ;
       sa[tmp[i]] = j ;
     }
     memcpy(rk , sa , n * sizeof(int)) ;
     memcpy(sa , tmp , n * sizeof(int)) ;
     if(j >= n - 1)  break ;
   }
   height[0] = 0 ;
   for(int i = 0 , k = 0 , j = rk[0] ; i < n - 1 ; i ++ , k ++)
     while(~k && s[i] != s[sa[j - 1] + k])
       height[j] = k -- , j = rk[sa[j] + 1] ;
}
int main()
{
   int n , k ;
   int num = 1e6 ;
   int ans = 0 , l = 1 , r = 0 ;
   scanf("%d%d" , &n , &k) ;
   for(int i = 0 ; i < n ; i ++)  scanf("%d" , &s[i]) ;
   if(n < k){printf("0\n") ; return 0 ;}
   suffixarray(n , num) ;
   for(int i = 2 ; i <= k ; i ++)
   {
   	  while(l <= r && height[q[r]] > height[i])  r -- ;
   	  q[++ r] = i ;
   }
   ans = max(ans , height[q[l]]) ;
   for(int i = k + 1 ; i <= n ; i ++)
   {
   	  while(l <= r && q[l] <= i - (k - 1))  l ++ ;
   	  while(l <= r && height[q[r]] > height[i])  r -- ;
   	  q[++ r] = i ;
   	  ans = max(ans , height[q[l]]) ;
   }
   printf("%d\n" , ans) ;
   return 0 ;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章