프로그래머스 - 테이블 해시 함수 (Java)

 

문제 개요

 

완호가 관리하는 어떤 데이터베이스의 한 테이블은 모두 정수 타입인 컬럼들로 이루어져 있습니다. 테이블은 2차원 행렬로 표현할 수 있으며 열은 컬럼을 나타내고, 행은 튜플을 나타냅니다.
첫 번째 컬럼은 기본키로서 모든 튜플에 대해 그 값이 중복되지 않도록 보장됩니다. 완호는 이 테이블에 대한 해시 함수를 다음과 같이 정의하였습니다.

1. 해시 함수는 col, row_begin, row_end을 입력으로 받습니다.
2. 테이블의 튜플을 col번째 컬럼의 값을 기준으로 오름차순 정렬을 하되, 만약 그 값이 동일하면 기본키인 첫 번째 컬럼의 값을 기준으로 내림차순 정렬합니다.
3. 정렬된 데이터에서 S_i를 i 번째 행의 튜플에 대해 각 컬럼의 값을 i로 나눈 나머지들의 합으로 정의합니다.
4. row_begin ≤ i ≤ row_end 인 모든 S_i를 누적하여 bitwise XOR 한 값을 해시 값으로서 반환합니다.

테이블의 데이터 data와 해시 함수에 대한 입력 col, row_begin, row_end이 주어졌을 때 테이블의 해시 값을 return 하도록 solution 함수를 완성해 주세요.

 

제한 사항

• 1 ≤ data의 길이 ≤ 2,500
• 1 ≤ data의 원소의 길이 ≤ 500
• 1 ≤ data[i][j] ≤ 1,000,000
  • data[i][j]는 i + 1 번째 튜플의 j + 1 번째 컬럼의 값을 의미합니다.
• 1 ≤ col ≤ data의 원소의 길이
• 1 ≤ row_begin ≤ row_end ≤ data의 길이

 

---

 

풀이

 

개인적으로 이 문제는 어떤 알고리즘을 사용해서 해결하는 게 목적이 아니라 다소 난해한 것처럼 보이는 문제 지문을 읽고, 그걸 코드로 작성할 수 있다면 간단하게 풀 수 있다고 생각한다.

 

지문을 읽고 작성할 코드의 순서를 대략적으로 생각해 봤다.

1. 2차원 배열인 int[][] data을 col번째 컬럼 값인 data[][col-1]을 기준으로 정렬한다.
   • 값이 같은 항목은 data[][0]을 기준으로 재정렬한다.
   • data[][0]은 지문에서 의미상으로 PK이기 때문에 중복되지 않는다.
2. data의 각 항목인 각각의 배열들에 대해 S_i를 구한다.
   • 예를 들어 S_2는 (data[2][0] % 2) + (data[2][1] % 2) + (data[2][2] % 2) + ...  이다.
3. 각 S_i들에 대해 모두 XOR시킨 결과를 리턴한다.

 

이번에 알게 된 사실이지만 Java로 XOR을 수행해야 하는 경우 ^ 연산자를 사용하면 쉽게 구할 수 있다.

 

---

 

코드

 

import java.util.*;

class Solution {
    public int solution(int[][] data, int col, int row_begin, int row_end) {
        Arrays.sort(data, (t1, t2) -> {
            if (t1[col-1] == t2[col-1]) {
                return t2[0] - t1[0];
            }
            return t1[col-1] - t2[col-1];
        });
        
        List<Integer> sums = new ArrayList<>();
        for (int i=row_begin-1 ; i<=row_end-1 ; i++) {
            int buf = 0;
            for (int n=0 ; n<data[i].length ; n++) {
                buf += data[i][n] % (i+1);
            }
            sums.add(buf);
        }

        int answer = sums.get(0);
        for (int i=1 ; i<sums.size() ; i++) {
            answer = answer ^ sums.get(i);
        }

        return answer;
    }
    
}

 

 

---

 

제출 결과

 

 

이번에는 문제를 풀 때 프로그래머스 사이트에서 코드를 작성하지 않고 인텔리제이 IDE에서 코드를 작성했다.

그리고 내가 인텔리제이에서 작성한 코드는 깃허브(여기 링크)에 올려두었다.

 

혹시 내용에 오류가 있거나 문제가 있다면 댓글 부탁드립니다!