[프로그래머스 Level 2] 테이블 해시 함수 (Java)

🚀링크

https://school.programmers.co.kr/learn/courses/30/lessons/147354

💻문제

문제 설명

완호가 관리하는 어떤 데이터베이스의 한 테이블은 모두 정수 타입인 컬럼들로 이루어져 있습니다. 테이블은 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의 길이

입출력 예
data col row_begin row_end result
[[2,2,6],[1,5,10],[4,2,9],[3,8,3]] 2 2 3 4

입출력 예 설명

• 정해진 방법에 따라 튜플을 정렬하면 {4, 2, 9}, {2, 2, 6}, {1, 5, 10}, {3, 8, 3} 이 됩니다.
• S_2 = (2 mod 2) + (2 mod 2) + (6 mod 2) = 0 입니다.
• S_3 = (1 mod 3) + (5 mod 3) + (10 mod 3) = 4 입니다.
• 따라서 해시 값은 S2 XOR S 3 = 4 입니다.

🌏문제 풀이

테이블의 튜플을 정렬하기 위해 int[]를 원소로 갖는 ArrayList를 선언하고, 데이터 베열에 있는 값을 저장합니다. 그 다음, 문제에 주어진 조건에 맞게 ArrayList를 정렬해주면 됩니다.

그 다음은 for문을 사용하여 각 튜플의 나머지들의 합을 계산하고, 재귀 호출을 하여 누적 bitwise XOR(^)을 계산하면 문제를 해결할 수 있습니다.

💻코드

import java.util.*;

class Solution {
    
    static int answer;
    static int[] S;
    static List<int[]> list;
    
    public int solution(int[][] data, int col, int row_begin, int row_end) {
        list = new ArrayList<>();
        for(int[] d : data) {
            list.add(d);
        }
        
        // 테이블 정렬
        Collections.sort(list, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                // 값이 동일하면 기본키인 첫 번째 컬럼의 값을 기준으로 내림차순 정렬
                if(o1[col - 1] == o2[col - 1]) {
                    return o2[0] - o1[0];
                }
                // col번째 컬럼의 값을 기준으로 오름차순 정렬
                return o1[col - 1] - o2[col - 1];
            }
        });
        
        int length = row_end - row_begin;
        S = new int[length + 1];

        // S_i를 i 번째 행의 튜플에 대해 각 컬럼의 값을 i 로 나눈 나머지들의 합
        calculate(row_begin, row_end);
        
        // 모든 S_i를 누적하여 bitwise XOR 한 값
        answer = bitwise(length);

        return answer;
    }
    
    private static void calculate(int start, int end) {
        for(int i = start; i <= end; i++) {
            for(int n : list.get(i - 1)) {
                S[i - start] += (n % i);
            }
        }
    }
    
    private static int bitwise(int k) {
        if(k == 0) {
            return S[k];
        }
        return bitwise(k - 1) ^ S[k];
    }
}