2025.4.16 알고리즘 코드카타

알고리즘 코드카타 - 122. 테이블 해시 함수

문제 링크

문제 설명

완호가 관리하는 어떤 데이터베이스의 한 테이블은 모두 정수 타입인 컬럼들로 이루어져 있습니다. 테이블은 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 입니다.
• 따라서 해시 값은 S2 XOR S 3 = 4 입니다.

풀이 코드

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int solution(vector<vector<int>> data, int col, int row_begin, int row_end) {
    int answer = 0;
    
    // col 전달을 위해 람다 사용자 정렬 함수 사용
    sort(data.begin(), data.end(), 
        [col](const vector<int>& a, const vector<int>& b) 
        {
            int i = col - 1;
            if (a[i] == b[i]) return a[0] > b[0];
            return a[i] < b[i];
        }
    );
    
    for(int i = row_begin - 1; i < row_end; i++)
    {
        int S_i = 0;
        for(int j = 0; j < data[i].size(); j++)
        {
            S_i += data[i][j] % (i+1);
        }
        answer ^= S_i; 
    }
    
    return answer;
}

• 로직 짜는 것보다 문제 설명을 이해하는 것이 더 어려웠던 문제..
• 컬럼은 그렇다쳐도 행은 튜플? 매개변수에선 row라고 쓰고 있잖아.
  
• 어느 체계에서 쓰느냐에 따라 달라지는 용어를 혼용해서 쓰고 있으니 헷갈릴 수밖에 없는 설명이었다.
• 문제 설명을 이해하고 나면 풀이 과정은 아래와 같이 단순화할 수 있다.

  1. 주어진 이차원행렬 data를 정렬한다.
  2. 정렬된 데이터 중, row_begin부터 row_end 사이의 행에 대하여 각 열의 S_i를 구한다.
  3. 구해진 S_i를 배타적 논리합(XOR) 연산으로 answer에 더한다.