버블 소트

개굴이·2023년 10월 5일
0

코딩테스트

목록 보기
40/58
post-thumbnail

백준 1517번 버블 소트

N개의 수로 이루어진 수열 A[1], A[2], …, A[N]이 있다. 이 수열에 대해서 버블 소트를 수행할 때, Swap이 총 몇 번 발생하는지 알아내는 프로그램을 작성하시오.

버블 소트는 서로 인접해 있는 두 수를 바꿔가며 정렬하는 방법이다. 예를 들어 수열이 3 2 1 이었다고 하자. 이 경우에는 인접해 있는 3, 2가 바뀌어야 하므로 2 3 1 이 된다. 다음으로는 3, 1이 바뀌어야 하므로 2 1 3 이 된다. 다음에는 2, 1이 바뀌어야 하므로 1 2 3 이 된다. 그러면 더 이상 바꿔야 할 경우가 없으므로 정렬이 완료된다.

입력

첫째 줄에 N(1 ≤ N ≤ 500,000)이 주어진다. 다음 줄에는 N개의 정수로 A[1], A[2], …, A[N]이 주어진다. 각각의 A[i]는 0 ≤ |A[i]| ≤ 1,000,000,000의 범위에 들어있다.

출력

첫째 줄에 Swap 횟수를 출력한다.

소스

import java.io.*;
import java.util.StringTokenizer;

public class Main {
    static long result = 0; //int형이 아닌 long형으로 선언해야 한다.

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        final int N = Integer.parseInt(br.readLine()); //숫자 개수
        int[] input = new int[N]; //입력받은 배열
        int[] tmp = new int[N]; //임시배열

        StringTokenizer st = new StringTokenizer(br.readLine());
        for(int i = 0; i < N; i++)
            input[i] = Integer.parseInt(st.nextToken());

        mergeSort(input, tmp, 0, N - 1);

        System.out.print(result);
    }

    static void mergeSort(int[] input, int[] tmp, int start, int end) {
        if (start < end) {
            int middle = (start + end) / 2;
            mergeSort(input, tmp, start, middle);
            mergeSort(input, tmp, middle + 1, end);
            merge(input, tmp, start, middle, end);
        }
    }
    static void merge(int[] arr, int[] tmp, int start, int middle, int end) {
        for(int i = start; i <= end; i++)
            tmp[i] = arr[i];

        int indexLeft = start;
        int indexRight = middle + 1;
        int index = start;

        while(indexLeft <= middle && indexRight <= end) {
            if(tmp[indexLeft] < tmp[indexRight])
                arr[index++] = tmp[indexLeft++];
            else {
                arr[index++] = tmp[indexRight++];
                result += middle - indexLeft + 1;
            }
        }
        if(indexLeft > middle)
            while(indexRight <= end)
                arr[index++] = tmp[indexRight++];
        else
            while(indexLeft <= middle)
                arr[index++] = tmp[indexLeft++];
    }

}

0개의 댓글