[프로그래머스] - 테이블 해시 함수(C/C++)

프로그래머스

[문제 설명]

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

---

[문제 풀이]

문제 내용대로 풀면 된다.

std::sort 함수의 정렬기준을 만들수 있는지 확인하는 문제라고 생각했다.

 

std::sort의 함수의 경우 기본적으로 컨테이너의 iterator.begin, iterator.end 두가지 파라미터만 전달하여 오름차순으로 정렬한다.

 

cpp reference 페이지에서 확인해보면 아래와 같이 3개의 파라미터를 전달하는 방식도 있다. 

하지만 std::sort의 사용법중에 사용자 함수를 넣어 사용자가 원하는대로 정렬 될 수 있도록 쓰는 방법이 있다.

대부분은 Compare 부분에 std::grearter<type>(), std::less<type>() 등과 같이 기본적으로 제공되는 Compare 함수를 사용한다.

 

사용자 함수를 만들어 사용하는 방법은 아래와 같다.

return 타입은 bool

const type &

로 전달하면 된다.

여기서 type은 컨테이너가 가지고 있는 변수를 넣어주면 된다.

만약 vector<int>라면, type은 int가 되고,

이번 문제와 같이 vector<vector<int>> 라면, type은 vector<int>가 된다.

 

해당 부분 헷갈리지 않도록 숙지해야하고, 까먹어도 포스트 맨 아래에 출처를 적어놓았으니 사이트에 들어가서 확인하면 된다.

 

함수 안의 내용은 간단하게 첫번째 파라미터 a, 두번째 파라미터 b일때

오름차순 -> return a < b

내림차순 -> return a > b

위와 같이 원하는 바를 return 해주면 된다.

 

문제를 풀면서 입력으로 주어진 col번째 원소를 기준으로 정렬해야 하는데,

우리가 사용하는 compare함수는 형식이 정해져있어서 col 값을 전달 할 수가 없다.

따라서 전역변수 gCol을 따로 두고, 입력받은 col 값을 전역변수에 옮겨서 전달하는 방법을 사용해야했다.

 

기준값인 각 튜플의 col값이 동일 할 경우에는 key값인 첫번째 원소를 기준으로 내림차순으로 정렬하면 된다. 

 

compare함수를 작성한 후 정상적으로 정렬이 되었다면 문제는 다 풀린것이라고 볼 수 있다.

 

모두 순회하며

 

각 튜플의 원소를 튜플의 인덱스 i로 mod(%)연산을 하여 전부 더해주고,

 

answer에 XOR(^) 연산을 해주면 된다.

[코드]

[GitHub]

GitHub - EGyeom/ProblemSolving

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

using namespace std;
int gCol = 0;
bool compare(const vector<int> &a, const vector<int> &b)
{
    if(a[gCol-1] == b[gCol-1])
        return a[0] > b[0];
    
    return a[gCol-1] < b[gCol-1];
}

int solution(vector<vector<int>> data, int col, int row_begin, int row_end) {
    int answer = 0;
    gCol = col;
    sort(data.begin(), data.end(), compare);

    for(int i = row_begin-1; i < row_end; i++)
    {
        int temp = 0;
        for(const int& n : data[i])
        {
            temp += n % (i+1);
        }
        answer ^= temp;
    }
    return answer;
}

 

출처

std::sort - cppreference.com