프로그래밍 공부
작성일
2024. 5. 15. 11:53
작성자
WDmil
728x90

문제 설명

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

 

제한 사항

  • 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

 


문제 해설

 

문제의 핵심을 파악해야 한다.

문제가 요구하는 사항은 해시값을 return 하라는것 이지만, 요구사항을 만족하기 위해 작성해야할 가장 중요한 알고리즘은 Sort 알고리즘 이다.

 

즉, 필요한 조건을 나열하면 다음과 같다.

  • 지정된 배열을 지정된 방식대로 정렬하라.
  • 배열의 결과값을 XOR 산출방식을 사용하여 범위값을 산출하라.

 

임의의 vector배열을 Sort하여 원하는 순서대로 정렬할 수 있어야 이 문제를 해결할 수 있다.

 

다양한 Sort방식을 찾아보면 도움이 된다.

 

마지막의 Result 호출방식에서 XOR로 결과값을 연산해야함을 명심하자.

https://inradestrt.tistory.com/633

 

Quick Sort 알고리즘

Quick Sort퀵정렬이란?찰스 앤터니 리처드 호어가 개발한 범용 정렬 알고리즘으로, 현 알고리즘 중 가장 고성능(최적의 항에서) 을 자랑하는 알고리즘으로, 주어진 배열을 정렬하는 분할 정복(Divide

inradestrt.tistory.com

 


첫 번째 시도

#include <string>
#include <vector>

using namespace std;

int partition(vector<vector<int>>& arr, int left, int right, int col) {
    vector<int> pivot = arr[right];
    int i = left - 1;

    for (int j = left; j < right; j++) {
        if (arr[j][col] < pivot[col]) {
            i++;
            swap(arr[i], arr[j]);
        }
        else if (arr[j][col] == pivot[col]) {
            if (arr[j][0] > pivot[0]) {
                i++;
                swap(arr[i], arr[j]);
            }
        }
    }
    swap(arr[i + 1], arr[right]);
    return i + 1;
}

void quickSort(vector<vector<int>>& arr, int left, int right, int col) {
    if (left < right) {
        int pivotIndex = partition(arr, left, right, col);
        quickSort(arr, left, pivotIndex - 1, col);  
        quickSort(arr, pivotIndex + 1, right, col);
    }
}

int solution(vector<vector<int>> data, int col, int row_begin, int row_end) {
    int answer = 0;
    quickSort(data, 0, data.size() - 1, col-1);
    for (int i = row_begin - 1; i < row_end; i++) {
        for (const int& def : data[i])
            answer += def % (i+1);
    }
    return answer;
}

실패

 

코드의 Sort과정은 정상이나 마지막 정답 도출과정에서 XOR처리를 해주지 않아서 실패했다.


두 번째 시도

#include <string>
#include <vector>

using namespace std;

int partition(vector<vector<int>>& arr, int left, int right, int col) {
    vector<int> pivot = arr[right];
    int i = left - 1;

    for (int j = left; j < right; j++) {
        if (arr[j][col] < pivot[col]) {
            i++;
            swap(arr[i], arr[j]);
        }
        else if (arr[j][col] == pivot[col]) {
            if (arr[j][0] > pivot[0]) {
                i++;
                swap(arr[i], arr[j]);
            }
        }
    }
    swap(arr[i + 1], arr[right]);
    return i + 1;
}

void quickSort(vector<vector<int>>& arr, int left, int right, int col) {
    if (left < right) {
        int pivotIndex = partition(arr, left, right, col);
        quickSort(arr, left, pivotIndex - 1, col);  
        quickSort(arr, pivotIndex + 1, right, col);
    }
}

int solution(vector<vector<int>> data, int col, int row_begin, int row_end) {
    int answer = 0;
    quickSort(data, 0, data.size() - 1, col-1);
    for (int i = row_begin - 1; i < row_end; i++) {
        int now = 0;
        for (const int& def : data[i])
            now += def % (i+1);
        answer ^= now;
    }
    return answer;
}

성공

 

마지막에 XOR처리만 진행해주면 성공이다.

728x90

'코딩테스트 문제 풀이 > 정렬' 카테고리의 다른 글

H-Index  (0) 2024.02.01
가장 큰 수  (0) 2024.01.23
K번째수  (0) 2024.01.12