본문 바로가기
알고리즘/라이브러리

Rabin karp

by 유시은 2021. 2. 10.
#include <bits/stdc++.h>
#define sad std::cout.flush(), system("pause")
using namespace std;
using ll = long long;
using pii = pair<int, int>;
using pll = pair<ll, ll>;

namespace rbk {
  const ll x = 131, md1 = 1e9+7, md2 = 1e9+9;
  int plen; pll d;

  void init(int n) {
    plen = n;
    d = {1, 1};
    for (int i = 0; i < n; ++i) {
      d.first = (d.first * x) % md1, d.second = (d.second * x) % md2;
    }
  }
  pll hash(string &s, int len = 0, int b = 0) {
    pll ret{0, 0};
    ll &r0 = ret.first, &r1 = ret.second;
    for (int i = 0; i < (len > 0 ? len : plen); ++i) {
      r0 = (r0 * x + s[b+i]) % md1, r1 = (r1 * x + s[b+i]) % md2;
    }
    return {r0, r1};
  }
  bool roll(pll &H, string &s, int p) {
    if (p + plen >= s.size()) return false;
    ll &h0 = H.first, &h1 = H.second;
    h0 = (h0 * x + s[p+plen]) % md1;
    h1 = (h1 * x + s[p+plen]) % md2;
    h0 = (h0 + (md1 - d.first * s[p] % md1) % md1) % md1;
    h1 = (h1 + (md2 - d.second * s[p] % md2) % md2) % md2;
    return true;
  }
}

 

해시 쌍으로 비교한다.

hash 즉시 $s_0 ... s_k$에 대한 해시가 반환됨에 유의.

 


공부한 곳

https://anz1217.tistory.com/62

 

'알고리즘 > 라이브러리' 카테고리의 다른 글

BIT  (0) 2021.03.22
DSU  (0) 2021.03.22
BFS  (0) 2020.12.13
Suffix Array와 LCP Array  (0) 2020.09.23
Topological Sort  (0) 2020.09.16

댓글