[백준] 1517번 버블 소트

문제

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

풀이

버블 소트에서 swap 되는 횟수를 세는 문제다.

문제가 버블 소트라고 해서 버블 소트로 구현을 한다면 시간초과가 나는 문제다.

버블 소트에서 swap 되는 조건을 이용하여 merge sort를 이용해야한다.

소스코드

#include <iostream>

using namespace std;

#define MAX_N 500000

int n;
int A[MAX_N];
int B[MAX_N];

long long solve(int start, int end)
{
    if (start == end) return 0;
    int mid = (start + end) / 2;

    long long ans = solve(start, mid) + solve(mid + 1, end);

    int i = start;
    int j = mid + 1;
    int k = 0;

    while (i <= mid || j <= end)
    {
        if (i <= mid && (j > end || A[i] <= A[j]))
        {
            B[k++] = A[i++];
        }
        else
        {
            // 왼쪽 배열의 남아있는 원소들의 개수
            ans += (long long)(mid - i + 1);
            B[k++] = A[j++];
        }
    }

    for (int i = start; i <= end; i++) {
        A[i] = B[i - start];
    }
    return ans;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);

    cin >> n;

    for (int i = 0; i < n; i++) cin >> A[i];

    long long ans = solve(0, n - 1);

    cout << ans;

    return 0;
}