CF 1968 G1.Division + LCP (easy version) (*1900) 二分+雜湊

Showball發表於2024-06-23

CF 1968 G1.Division + LCP (easy version) (*1900) 二分+雜湊

題目連結

題意：

給你一個字串 \(s\) , 請你把字串分割成 \(k\) 份。使得這些字串的最長公共字首的長度最大。

思路：

最長公共字首的長度具有單調性，因此可以進行二分。考慮如何check。我們可以貪心的去找出含有公共字首的一段分成一組，然後比較分出的組數是否大於等於 \(k\) 即可。至於如何找出公共字首可以用字串雜湊解決。

注意雜湊陣列開兩倍空間，防止RE。

程式碼：

#include<bits/stdc++.h>

using namespace std;

#define ff first
#define ss second
#define pb push_back
#define all(u) u.begin(), u.end()
#define endl '\n'
#define debug(x) cout<<#x<<":"<<x<<endl;

typedef pair<int, int> PII;
typedef long long LL;
const int inf = 0x3f3f3f3f;
const int N = 1e5 + 10, M = 105;
const int mod = 1e9 + 7, mod1 = 1e9+9;
const int P1=131,P2=13331;
const int cases = 1;

void Showball(){
   int n,k;
   cin>>n>>k>>k;
   string s;
   cin>>s;
   s="?"+s; 
   vector<LL> p1(2*n+1),p2(2*n+1),h1(2*n+1),h2(2*n+1);
   p1[0]=p2[0]=1;
   for(int i=1;i<=n;i++){
      p1[i]=(p1[i-1]*P1)%mod;
      p2[i]=(p2[i-1]*P2)%mod1;
      h1[i]=((h1[i-1]*P1)%mod+s[i]-'0'+1)%mod;
      h2[i]=((h2[i-1]*P2)%mod1+s[i]-'0'+1)%mod1;
   }
   auto get1=[&](int l,int r){return (h1[r]-(h1[l-1]*p1[r-l+1])%mod+mod)%mod;};
   auto get2=[&](int l,int r){return (h2[r]-(h2[l-1]*p2[r-l+1])%mod1+mod1)%mod1;};

   auto check=[&](int mid){
      if(!mid) return true;
      int cnt=1;
      int hash1=get1(1,mid);
      int hash2=get2(1,mid);
      int i=mid+1;
      while(i<=n){
         if(get1(i,i+mid-1)==hash1&&get2(i,i+mid-1)==hash2) cnt++,i+=mid;
         else i++;
      }
      return cnt>=k;
   };

   int l=0,r=n;
   while(l<r){
      int mid=(l+r+1)>>1;
      if(check(mid)) l=mid;
      else r=mid-1;
   }
   cout<<l<<endl;
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int T=1;
    if(cases) cin>>T;
    while(T--)
    Showball();
    return 0;
}

相關文章