본문 바로가기

Algorithm/Problems

[JAVA]Hashing(Baekjoon)

문제

문제

 APC에 온 것을 환영한다. 만약 여러분이 학교에서 자료구조를 수강했다면 해시 함수에 대해 배웠을 것이다. 해시 함수란 임의의 길이의 입력을 받아서 고정된 길이의 출력을 내보내는 함수로 정의한다. 해시 함수는 무궁무진한 응용 분야를 갖는데, 대표적으로 자료의 저장과 탐색에 쓰인다.

 이 문제에서는 여러분이 앞으로 유용하게 쓸 수 있는 해시 함수를 하나 가르쳐주고자 한다. 먼저, 편의상 입력으로 들어오는 문자열에는 영문 소문자(a, b, ..., z)로만 구성되어있다고 가정하자. 영어에는 총 26개의 알파벳이 존재하므로 a에는 1, b에는 2, c에는 3, ..., z에는 26으로 고유한 번호를 부여할 수 있다. 결과적으로 우리는 하나의 문자열을 수열로 변환할 수 있다. 예를 들어서 문자열 "abba"은 수열 1, 2, 2, 1로 나타낼 수 있다.

 해시 값을 계산하기 위해서 우리는 문자열 혹은 수열을 하나의 정수로 치환하려고 한다. 간단하게는 수열의 값을 모두 더할 수도 있다. 해시 함수의 정의에서 유한한 범위의 출력을 가져야 한다고 했으니까 적당히 큰 수 M으로 나눠주자. 짜잔! 해시 함수가 완성되었다. 이를 수식으로 표현하면 아래와 같다.
\begin{align}
H= \sum_{i=0}^{l-1} a_i \quad\bmod M
\end{align}
 해시 함수의 입력으로 들어올 수 있는 문자열의 종류는 무한하지만 출력 범위는 정해져있다. 다들 비둘기 집의 원리에 대해서는 한 번쯤 들어봤을 것이다. 그 원리에 의하면 서로 다른 문자열이더라도 동일한 해시 값을 가질 수 있다. 이를 해시 충돌이라고 하는데, 좋은 해시 함수는 최대한 충돌이 적게 일어나야 한다. 위에서 정의한 해시 함수는 알파벳의 순서만 바꿔도 충돌이 일어나기 때문에 나쁜 해시 함수이다. 그러니까 조금 더 개선해보자.

 어떻게 하면 순서가 달라졌을때 출력값도 달라지게 할 수 있을까? 머리를 굴리면 수열의 각 항마다 고유한 계수를 부여하면 된다는 아이디어를 생각해볼 수 있다. 가장 대표적인 방법은 항의 번호에 해당하는 만큼 특정한 숫자를 거듭제곱해서 곱해준 다음 더하는 것이 있다. 이를 수식으로 표현하면 아래와 같다.
\begin{align}
H= \sum_{i=0}^{l-1} a_ir^i \quad\bmod M
\end{align}
 보통 r과 M은 서로소인 숫자로 정하는 것이 일반적이다. 우리가 직접 정하라고 하면 힘들테니까 r의 값은 26보다 큰 소수인 31로 하고 M의 값은 1234567891(놀랍게도 소수이다!!)로 하자.

 이제 여러분이 할 일은 위 식을 통해 주어진 문자열의 해시 값을 계산하는 것이다. 그리고 이 함수는 간단해 보여도 자주 쓰이니까 기억해뒀다가 잘 써먹도록 하자.

입력

첫 줄에는 문자열의 길이 L이 들어온다. 둘째 줄에는 영문 소문자로만 이루어진 문자열이 들어온다.

입력으로 주어지는 문자열은 모두 알파벳 소문자로만 구성되어 있다.

출력

문제에서 주어진 해시함수와 입력으로 주어진 문자열을 사용해 계산한 해시 값을 정수로 출력한다.

Small (50점)

  • $1 \le L \le 5$

Large (50점)

  • $1\le L \le 50$

예제 입력 1

5
abcde

예제 출력 1

4739715

예제 입력 2

3
zzz

예제 출력 2

25818

예제 입력 3

1
i

예제 출력 3

9

풀이

 처음에는 본 문제를 풀기 위해 Math.pow 함수를 사용하여 제곱을 계산하고 반복문을 통해 하나의 변수에 값을 더해주었다. 그렇게하니 50점이 나왔다.
 자료형의 문제라고 생각을 하여 int를 long으로 바꾸어 진행을 하였다. 그래도 50점이 나왔다.
 결국 구글링을 하여 찾아보았는데, Large의 경우 L이 50까지 입력되는데 그렇게 되면 31의 49제곱이 계산되고, 이는 overflow가 발생하여 틀린 값으로 나오는 것이었다.
 따라서 이 문제는 모듈러 연산의 성질을 이용하여 풀어야 한다. 모듈러 연산은 분배법칙이 가능하다.

\begin{align}
H
&= \sum_{i=0}^{l-1} a_ir^i \quad\bmod M \\\
&= (a_0r^0 + a_1r^1 + a_2r^2 + \cdots + a_{l-1}r^{l-1}) \bmod M \\\
&= (a_0r^0\bmod M + a_1r^1\bmod M + a_2r^2\bmod M + \cdots + a_{l-1}r^{l-1}\bmod M) \bmod M \\\
\end{align}

 이렇게 코드를 작성해도 overflow가 발생할 수 있기 때문에, 곱셈의 분배법칙도 적용해야 한다.

\begin{align}
a_2r^2\bmod M &= (a_2\bmod ,*,, r^2\bmod M) \bmod M \\\
r^2\bmod M &= (r\bmod *,, r\bmod M) \bmod M \\\
r^3\bmod M &= (r^2\bmod *,, r\bmod M) \bmod M \\\
r^4\bmod M &= (r^3\bmod *,, r\bmod M) \bmod M \\\
\end{align}

public class _15829 {
    static final int M = 1234567891;    // 상수 M
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int L = Integer.parseInt(br.readLine());
        String str = br.readLine();
        int[] arr = new int[L];
        long sum = 0;
        long pow = 1;
        for(int i = 0; i < L; i++){
            arr[i] = str.charAt(i) - 'a' + 1;
        }
        for(int i = 0; i < L; i++){
            sum += arr[i] * pow % M;    //분배 법칙
            pow = pow * 31 % M;         //분배 법칙
        }
        System.out.println(sum % M);
    }
}

결론

자료형에 따라 값의 범위가 한정된다. 그리고, 여러 연산을 사용하는 해싱 등의 방식에서는 overflow가 발생하기 쉽다. 따라서 자주 사용되는 수학적 법칙들은 익혀서 연산의 최적화나 값의 최적화를 체화할 필요가 있다고 생각했다.

'Algorithm > Problems' 카테고리의 다른 글

[JAVA]앱(Baekjoon)  (0) 2023.02.15
[JAVA]수 묶기(Baekjoon)  (0) 2023.02.15
[JAVA]이항 계수 1(Baekjoon)  (0) 2023.02.15
[JAVA]숫자 카드 2(Baekjoon)  (0) 2023.02.15
[JAVA]수 찾기(Baekjoon)  (0) 2023.02.15