[프로그래머스 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 입니다.
• 따라서 해시 값은 S_2 XOR S_ 3 = 4 입니다.

 

 

 

 

 

 

나의 코드

import java.util.*;

class Solution {
   
   public int solution(int[][] data, int col, int row_begin, int row_end) {
        int answer = 0;
        int[] S_i = new int[row_end - row_begin + 1];
        int idx = 0;
        
        Arrays.sort(data, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                if(o1[col-1] == o2[col-1])
                    return o2[0] - o1[0];
                
                return o1[col-1] - o2[col-1];
            }
        });
        
        for(int i=row_begin-1; i<=row_end-1; i++) {
            for(int j=0; j<data[0].length; j++) {
                S_i[idx] += data[i][j] % (i+1);
            }
            // System.out.println("S_i[" + idx + "] = " +  S_i[idx]);
            idx++;
        }
        
        int temp = S_i[0];
        for(int i=1; i<S_i.length; i++) {
            temp = temp ^ S_i[i]; // XOR 연산자(^)를 통해 XOR 연산 가능
        }
        
        answer = temp;
        
        return answer;
    }
}

 

풀이

1. 문제에 나온대로 순서대로 2차원 배열 data를 정렬한 후에 S_i를 구하고 S_i를 누적하여 XOR 연산을 진행하면 된다.
2. 우선 Comparator 인터페이스를 이용하여 compare 메소드를 구현하여  2차원 배열 data를 정렬한다. data[i]의 col번 째 칼럼으로 오름차순 정렬하고 data[i]의 col번 째 칼럼이 같은 경우 data[i]의 첫번 째 칼럼으로 내림차순 정렬한다.  index는 0부터 시작하므로 col - 1번 째로 계산하여 정렬한다.
3. 그리고 정렬한 data를 돌면서 S_i 값들을 구하면 된다.
4. S_i를 누적해가며 XOR 연산을 진행할 것이므로 row_begin부터 row_end 까지만 구해서 S_i의 index 0부터 값을 넣어주도록 한다. 따라서 row_end - row_begin + 1 크기의 배열 S_i를 생성한다.
5. row_begin과 row_end도 순서이므로 index 계산할 때 주의하여 인덱스 data의 row_begin - 1 부터 row_end - 1 행까지S_i를 구하여 순서대로 S_i의 0 인덱스부터 채운다.
6. 이제 S_i를 누적하며 XOR 연산을 진행하면 된다. int형 변수 temp를 선언하고 S_i의 0인덱스 값을 저장하고 S_i의 1인덱스부터 for문을 돌려가며 XOR 연산을 하고 temp에 저장하고 다시 temp와 S_i 다음 인덱스 값과 XOR 연산을 진행한다.
7. 최종 XOR 연산을 완료한 값을 리턴한다.