[C++] sort 함수 사용하기

알고리즘 대회나 코딩테스트에서 요소를 오름차순, 혹은 내림차순으로 정렬해야하는 상황이 자주 주어진다. 직접 정렬을 구현할 수 있다면 문제되지않으나 문제를 풀 때마다 구현하기가 쉽지않으며, 시간도 많이 소요된다. 

 

다행히 C++ 에선 퀵 정렬을 기반으로한 sort함수를 제공한다. 기본적으로 퀵 정렬은 worst case 에서 N^2 의 시간복잡도를 가지기 때문에 항상 사용하기에 무리가 있지만, sort함수는 퀵 정렬 +  힙 정렬 + 삽입 정렬을 섞은 intro sort방식으로 구현되어 worst case 에서도 nlogn의 복잡도를 가진다.

 

 

배열과 벡터에서 sort를 사용하는 방법은 다음과 같다

 

오름차순

#include<algorithm>

int arr[10] = {5, 2, 10, 9, 1, 3, 4, 8, 7, 6};
sort(arr, arr + 10);

vector<int> v = {5, 2, 10, 9, 1, 3, 4, 8, 7, 6};
sort(v.begin(), v.end());

결과

1 2 3 4 5 6 7 8 9 10 1 2 3 4 5 6 7 8 9 10

시작위치와 끝 위치를 지정해주면 된다. 기본적으로 오름차순으로 정렬한다. 그렇다면 내림차순으로 정렬하고 싶을 땐 어떻게 해야할까? boolean 형식의 사용사 함수를 만들어 sort 함수에 인자로 넣어주면 된다.

 

내림차순 

#include<algorithm>

bool compare(int i, int j){
    return i > j;
}

// ... 생략

sort(arr, arr + 10, compare);
sort(v.begin(), v.end(), compare);

결과

10 9 8 7 6 5 4 3 2 1 10 9 8 7 6 5 4 3 2 1

 

 

코드만으론 이해가 부족할 수 있으니 직접 문제를 풀어보면 좋을것 같았다. 아래는 백준알고리즘과 풀이 링크이다.

 

https://www.acmicpc.net/problem/11650

11650번: 좌표 정렬하기

https://infjin.tistory.com/5?category=992230 

백준알고리즘 11650 좌표정렬하기 풀이 C++