[백준 1517] 버블 소트 (JAVA)
문제
풀이
버블 정렬을 이용하면 n^2으로 시간초과가 발생합니다. 병합 정렬을 이용하여 nlogn으로 해결할 수 있습니다.
- 병합 정렬을 진행하면서 두 개의 배열을 병합할 때 Swap 횟수를 계산할 수 있습니다.
- RightArray 값이 작으므로 1를 넣습니다. RightArray의 값이 들어가는 경우 현재 LeftArray의 고르지 않은 값들의 개수(Count)를 Swap에 더합니다.
- LeftArray 값이 작으므로 2를 넣습니다. LeftArray의 값이 들어가는 경우 Count를 1 줄입니다.
- LeftArray 값이 작으므로 3을 넣고 Count를 줄입니다.
- RightArray 값이 작으므로 4를 넣고 Swap에 Count를 더합니다.
- 한쪽 배열이 비어있으면 다른 배열의 값을 그냥 넣습니다.
모든 병합을 계산하면 버블 정렬의 Swap 개수를 찾을 수 있습니다.
코드
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;
public class Main {
public static int N;
public static int[] arr;
public static long ans = 0;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
N = Integer.parseInt(br.readLine());
arr = new int[N];
StringTokenizer st = new StringTokenizer(br.readLine());
for (int i = 0; i < N; i++) {
arr[i] = Integer.parseInt(st.nextToken());
}
mergeSort(0, arr.length - 1);
System.out.println(ans);
br.close();
}
public static void mergeSort(int left, int right) {
if (left < right) {
int mid = (left + right) >> 1;
mergeSort(left, mid);
mergeSort(mid + 1, right);
merge(left, mid, right);
}
}
public static void merge(int left, int mid, int right) {
int[] leftArr = Arrays.copyOfRange(arr, left, mid + 1);
int[] rightArr = Arrays.copyOfRange(arr, mid + 1, right + 1);
int index = left, leftIdx = 0, rightIdx = 0, cnt = leftArr.length;
while (leftIdx < leftArr.length && rightIdx < rightArr.length) {
if (leftArr[leftIdx] > rightArr[rightIdx]) {
arr[index] = rightArr[rightIdx++];
ans += cnt;
} else {
arr[index] = leftArr[leftIdx++];
cnt--;
}
index++;
}
while (rightIdx < rightArr.length) {
arr[index++] = rightArr[rightIdx++];
}
while (leftIdx < leftArr.length) {
arr[index++] = leftArr[leftIdx++];
}
}
}
Author And Source
이 문제에 관하여([백준 1517] 버블 소트 (JAVA)), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@solser12/백준-1517-버블-소트-JAVA저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)