[프로그래머스] 테이블 해시 함수

문제

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

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

풀이

문제를 잘 읽으면 문제의 흐름대로 풀면 된다.
제일 처음에 해야 될 것은 정렬이다.
Arrays도 좋지만 List의 사용이 더 이점이 있으므로 List를 가지고 있는 static class를 만들어서 이를 저장할 수 있도록 하였다.
이렇게 담은 List에서 정렬기준을 Comparotor를 이용하여 재정의 한다.
col값에 따라 내림차순, 첫 인덱스에 따라 오름차순이다.

이렇게 정렬한 리스트에서 i번째 행에서 mod값을 구한 후 각 Sum에서 XOR연산 처리를 해준다.

import java.util.*;
import java.lang.*;

class Solution {
    public int solution(int[][] data, int col, int row_begin, int row_end) {
        int answer = 0;
        List<Tuple> tupleList = new ArrayList<>();
        for(int i = 0; i < data.length; i++) {
            tupleList.add(new Tuple(data[i]));
        }

        Collections.sort(tupleList, new Comparator<Tuple>() {
            public int compare(Tuple a, Tuple b) {
                if (a.list.get(col -1) < b.list.get(col -1)) {
                    return -1;
                } else if (a.list.get(col -1) > b.list.get(col -1)) {
                    return 1;
                } else {
                    if (a.list.get(0) < b.list.get(0)) {
                        return 1;
                    } else {
                        return -1;
                    }
                }
            }
        });

        for (int i = row_begin; i <= row_end; i++) {
            Tuple tuple = tupleList.get(i - 1);
            List<Integer> list = tuple.getList();
            int sum = 0;
            for(int a : list) {
                sum += a % i;
            }

            answer ^= sum;
        }

        return answer;
    }

    static class Tuple {
        private List<Integer> list = new ArrayList<>();

        public Tuple(int[] data) {
            for(int a : data) {
                list.add(a);
            }
        }

        public List<Integer> getList() {
            return list;
        }
    }
}

https://github.com/Win-9/Algorism/tree/main/programers/%ED%85%8C%EC%9D%B4%EB%B8%94%20%ED%95%B4%EC%8B%9C%20%ED%95%A8%EC%88%98