Advertisement
Guest User

Google Kickstart Round E Problem C Hashing based solution

a guest
Aug 21st, 2022
307
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.11 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. #include <ext/pb_ds/assoc_container.hpp>
  3. #include <ext/pb_ds/tree_policy.hpp>
  4. #define int long long
  5. #define pb push_back
  6. #define ff first
  7. #define ss second
  8. const int N=3e5+7;
  9. const long long mod=1e9+7;
  10. const double eps=1e-6;
  11. const double pi=    3.14159265358979323846;
  12. using namespace std;  
  13. using namespace __gnu_pbds;
  14. typedef tree<int, null_type, less_equal<int>, rb_tree_tag,
  15.              tree_order_statistics_node_update>
  16.     op_set;
  17.  
  18. long long binpow(long long base,long long ex)
  19. {
  20.     long long ans=1ll;
  21.     while(ex>0)
  22.     {
  23.         if(ex%2==1)
  24.             ans=(ans*base)%mod;
  25.         base=(base*base)%mod;
  26.         ex=ex/2;
  27.     }
  28.     return ans;
  29. }
  30.  
  31. int32_t main()
  32. {
  33.    ios_base::sync_with_stdio(false);
  34.    cin.tie(0);
  35.  
  36.    int t;
  37.    cin >> t;
  38.  
  39.    for(int tc=1;tc<=t;tc++)
  40.    {
  41.         cout << "Case #" << tc << ": ";
  42.  
  43.         int n;
  44.         string s;
  45.         cin >> n >> s;
  46.  
  47.         vector<long long> pre(n+2);
  48.         vector<long long> suf(n+2);
  49.         vector<long long> power(n+2);
  50.         power[0]=1;
  51.         for(int i=1;i<=n;i++)
  52.                 power[i]=(power[i-1]*31ll)%mod;
  53.  
  54.         pre[0]=0;
  55.         for(int i=1;i<=n;i++)
  56.         {
  57.             long long rep=s[i-1]-'a'+1;
  58.             pre[i]=(rep*power[i]+pre[i-1])%mod;
  59.         }
  60.  
  61.         suf[n+1]=0;
  62.         for(int i=n;i>=1;i--)
  63.         {
  64.             long long rep=s[i-1]-'a'+1;
  65.             suf[i]=(rep*power[n+1-i]+suf[i+1])%mod;
  66.         }      
  67.  
  68.         int ans=n;
  69.         for(int len=1;len<=n;len++)
  70.         {
  71.             long long p1=pre[len];
  72.             long long p2=( (suf[1]-suf[len+1]+mod)%mod * binpow(power[n-len],mod-2) )%mod;
  73.  
  74.             if(p1==p2)
  75.             {
  76.                 long long s1=suf[len+1];
  77.                 long long s2=( (pre[n]-pre[len]+mod)%mod * binpow(power[len],mod-2) )%mod;
  78.  
  79.                 if(s1==s2)
  80.                 {
  81.                     ans=len;
  82.                     break;
  83.                 }
  84.             }
  85.         }
  86.  
  87.         cout << s.substr(0,ans) << "\n";
  88.     }
  89. }
  90.  
  91.  
  92. /*
  93. https://www.youtube.com/watch?v=9pV5HjeVrAg
  94. Just Wow.
  95. */
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement