[프로그래머스] 테이블 해시 함수 Lv2 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 함수를 완성해 주세요.

해설

문제만 잘 읽어보면 쉽게 풀 수 있는 문제라고 생각하여 해설이 필요 없을 수 있지만, 그래도 모를 수 있는 내용이 있기 때문에 포스팅을 했습니다.

문제를 풀기 전에 포인트만 짚고 넘어가겠습니다.

• col 번째 컬럼(col - 1번째 인덱스)을 기준으로 오름차순 정렬하고 값이 동일하다면, 첫 번째 컬럼(0번째 인덱스) 기준으로 내림차순 정렬 합니다.
• mod는 나누기의 나머지 값을 뜻합니다.
• XOR 연산자는 Java에서 '^'로 계산할 수 있습니다.

위의 3가지 정도 알고 문제만 잘 읽어보면 쉽게 풀어 나갈 수 있다고 생각합니다!

 

코드

import java.util.*;

class Solution {
    public int solution(int[][] data, int col, int row_begin, int row_end) {
        int answer = 0;
        
        Arrays.sort(data, ((x, y) -> {
            if (x[col - 1] == y[col - 1]) {
                return y[0] - x[0];
            }
            
            return x[col - 1] - y[col - 1];
        }));
        
        for (int i = row_begin; i <= row_end; i++) {
            answer ^= calcMod(data[i - 1], i);
        }
        
        return answer;
    }
    
    public int calcMod(int[] items, int mod) {
        int answer = 0;
        
        for (int item: items) {
            answer += item % mod;
        }
        
        return answer;
    }
}