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
첫 번째 시도
#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 |