Merge Sort

merge sort를 이용하여 inversion 개수세기

[1517 버블 소트] https://www.acmicpc.net/problem/1517 .
inversion의 개수를 센다.

문제상황 파악하기.

버블소트는 arr[i]>arr[i+1]이면 swap하면서 진행하는 정렬 방법이다.
그리고 이는 당연하게도 O(n^2)이 걸린다.
그리고 당연하게도 그대로 구현하면 시간초과다 그러면 실버급이겠지.
O(NlogN)까지는 허용 될 것이므로 다른 방법을 찾아야한다.

inversion?

i < j 일 때 arr[i] > arr[i+1] 일때 (arr[i], arr[i+1])의 쌍을 inversion이라고 한다.
결국 이 문제는 모든 inversion의 개수를 세는 것이다.
inversion을 세는 방법은 merge sort, bit, segment tree 등 여러가지가 있지만 merge sort를 이용해 구해보자!
[참고자료] https://www.geeksforgeeks.org/counting-inversions-in-an-array-using-segment-tree/?ref=gcse
나는 위의 링크를 참고 했다.

아이디어 얻기.

merge sort는 Divide & Conquer를 이용한다.
분해되고 병합되는 과정은 아래 그림과 같다.
problem D
최대한 절반으로 나눠서 효율적인 계산을 하도록 하고,
노란색, 초록색으로 분해 되고 병합된다. 그 과정에서 빨간색 박스로 inversion을 센다.
병합하는 과정에서 mid-i개(앞인데 더 큰거의 개수)를 계속 더해주는 것이다.

주의할 점

값이 long long을 넘어갈 수 있다. 범위를 보면 쉽게 알 수 있다.
(근데 난 한번 틀렸다 제출하자마자 아차 했다 ㅋㅋ)

실제 코드

나머지 주의 할 점은 코드에 주석으로 처리했다.

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
#include <queue>
#include <cmath>
#include <set>
#define fast_io cin.tie(NULL); cout.tie(NULL); ios_base::sync_with_stdio(false);
using namespace std;
typedef long long ll;

int n;
ll arr[500001];

ll merge(ll arr[], ll temp[], int left, int mid, int right){
    int i, j, k;
    ll inv_count = 0;
    i = left; j = mid; k = left;
    while (i<mid&&j<=right){
        if(arr[i]<=arr[j]){
            temp[k++] = arr[i++];
        }else{
            temp[k++] = arr[j++];
            //mid는 right덩어리의 처음 index이다. 
            inv_count += mid-i;
        }
    }
    //남은거 다 넣어
    while (i<mid) temp[k++] = arr[i++];
    while (j<=right) temp[k++] = arr[j++];
    //원래 배열로 다 옮겨
    for(i= left;i<=right;i++) arr[i] = temp[i];
    return inv_count;
}

ll _mergeSort(ll arr[], ll temp[], int left, int right){
    int mid; ll inv_count=0;
    if(right>left){
        mid = (left+right)>>1;
        inv_count += _mergeSort(arr, temp, left, mid);
        inv_count += _mergeSort(arr, temp, mid+1, right);

        inv_count += merge(arr, temp, left, mid+1, right);
    }
    return inv_count;
}

ll mergeSort(ll arr[], int array_size){
    ll temp[array_size];
    return _mergeSort(arr, temp, 0, array_size-1);  //mergesort call-by-address
}

int main(){
    fast_io;
    cin >> n;
    for(int i=0;i<n;i++) cin >> arr[i];
    cout << mergeSort(arr, n);
}

Leave a comment

2024

D&C Optimization

3 minute read

분할정복을 이용한 다이나믹 프로그래밍 최적화

Back to top ↑

2023

Back to top ↑

2022

Greedy

2 minute read

백준 18186번 라면사기(large) 문제풀이

SCC

1 minute read

백준 4196번 도미노 문제풀이

LCA

2 minute read

백준 3176번 도로 네트워크 문제풀이

2-SAT

1 minute read

백준 16367번 TV Show Game 문제풀이

Hashing

2 minute read

백준 21162번 뒤집기 K 문제풀이

Tree DP

1 minute read

백준 1949번 우수 마을 문제풀이

Trie

1 minute read

Trie 자료구조 이해하기

Merge Sort

1 minute read

merge sort를 이용하여 inversion 개수세기

Segment Tree

2 minute read

백준 11505 구간 곱 구하기 문제풀이

BFS

1 minute read

백준 2206 벽 부수고 이동하기 문제풀이

기하

1 minute read

백준 2166 다각형의 면적 문제풀이

이분탐색

1 minute read

백준 12015 가장 긴 증가하는 부분 수열2 문제풀이

누적 합

1 minute read

백준 10986 나머지합 문제풀이

Stack

1 minute read

스택 구현하기 ========== 자료구조의 기본이라고 하면 스택 과 큐가 있다 백준 10828번에서 마주친 스택

정렬1

1 minute read

매우매우 많은 정렬이 있지만 그중 가장 안 어려운 3가지를 공부해보았다.

Back to top ↑