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

문제출처
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의 길이

입출력 예

문제풀이

1. 정렬한 다음 주어진 구간만큼의 배열을 계산하여 XOR을 계산한 값을 정답으로 리턴한다.

• 문제에 나온 순서대로 계산해주면 된다.

코드

function solution(data, col, row_begin, row_end) {
    var answer = 0;
    
    data.sort((a,b) => {
        if(a[col-1] == b[col-1]) return b[0] - a[0];
        else return a[col-1] - b[col-1];
    })
    
    let arr = [];
    data.map((v,i) => {
        let sum = 0;
        if(i+1 >= row_begin && i+1 <= row_end){
            v.map(e => {
                sum += e%(i+1);
            })
            
            answer = answer ^ sum;
        }
    })
    
    return answer;
}

리팩토링

function solution(data, col, row_begin, row_end) {
    var answer = 0;
    data.sort((a,b) => a[col-1] - b[col-1] || b[0]-a[0]).map((v,i) => {
        if(i+1 >= row_begin && i+1 <= row_end)  
            answer ^= v.reduce((acc, cur) => acc + cur%(i+1),0);
    })
    return answer;
}

다른 사람 코드

function sort(a, b, col) {
    return a[col - 1] - b[col - 1] || b[0] - a[0];
}

function solution(data, col, row_begin, row_end) {
    data.sort((a, b) => sort(a, b, col));
    return data
        .map((row, i) => row.reduce((acc, col) => acc + (col % (i + 1)), 0))
        .slice(row_begin - 1, row_end)
        .reduce((acc, val) => acc ^ val, 0);
}

TIL

• sort((a,b) => a[0] - b[0] || b[0] - a[0]) : 앞 조건의 값이 동일하다면 뒷 조건으로 정렬
• XOR : ^으로 계산 가능