[프로그래머스] Swift - 테이블 해시 함수 (LV.2)

👆문제풀러 가기👆

문제 설명

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

---

풀이 [ 메모리: 17mb, 최대시간: 2.17ms ]

 

func solution(_ data:[[Int]], _ col:Int, _ row_begin:Int, _ row_end:Int) -> Int {
    // 기본키를 Key로 설정, 각 Tuple을 Value로 설정
    let copyData = makeCopyData(of: data)

    // 정렬기준 - col번째 열 기준으로 오름차순
    //        - 값이 같은 게 있으면 기본키(col[0]) 기준으로 내림차순
    let sortedTable = sortTuple(copyData, col)
    var result = 0

    // row_begin ~ row_end 범위 내의 인덱스만 Tuple 확인\
    // Tuple(배열) 내의 값 합산
    // 각각의 값을 XOR연산
    for row in 0..<sortedTable.count where row_begin...row_end ~= (row+1) {
        result ^= sortedTable[row].value.map{$0 % (row+1)}.reduce(0, +)
    }

    return result
}

func makeCopyData(of data: [[Int]]) -> [Int: [Int]] {
    var copyData = [Int: [Int]]()

    // 기본키를 Key로 설정, 각 Tuple을 Value로 설정
    for row in 0..<data.count {
        let defaultKey = data[row][0]
        copyData[defaultKey] = data[row]
    }
    return copyData
}

func sortTuple(_ data: [Int: [Int]], _ col: Int) -> [(key: Int, value: [Int])] {
    return data.sorted {
        if $0.value[col-1] == $1.value[col-1] {
            return $0.key > $1.key
        }
        return $0.value[col-1] < $1.value[col-1]
    }
}

아마도 비트연산을 모르면 풀 수 없는 문제.

문제가 괜히 복잡하게 적혀있어 이해하는 데 시간이 조금 걸렸지만, 어렵지 않은 문제이다.

 

고려사항

1. 정렬의 기준(col번째 열 기준으로 오름차순, 동일한 값이 있으면 기본키(각 row의 첫번째 값) 기준으로 정렬)
   • 그렇다면, 2차원 배열을 row단위로 어떻게 정렬할 수 있을까?를 고민해야한다.
2. S는 정렬된 테이블, S_i는 정렬된 테이블의 row(i번째 행)을 의미
3. 비트연산(XOR연산) 기호

 

정렬 조건이 특이했고, 풀이도 지금까지 풀어보지 않은 방식이라 개인적으로 재밌었던 문제이다.

한 함수에 10줄 이상이면 읽기 불편해서 함수를 최대한 빼려고 노력하는데, 잘 하고 있는건지 모르겠다.