알고리즘 수업 - 삽입 정렬 1
시간 제한1초메모리 제한512 MB
주어진 삽입 정렬 의사 코드를 그대로 실행해 K번째로 저장되는 값을 출력하고, 저장 횟수가 K보다 적으면 -1을 출력합니다.
문제
오늘도 서준이는 삽입 정렬 수업 조교를 하고 있다. 아빠가 수업한 내용을 학생들이 잘 이해했는지 문제를 통해서 확인해보자.
N개의 서로 다른 양의 정수가 저장된 배열 A가 있다. 삽입 정렬로 배열 A를 오름차순 정렬할 경우 배열 A에 K 번째 저장되는 수를 구해서 우리 서준이를 도와주자.
크기가 N인 배열에 대한 삽입 정렬 의사 코드는 다음과 같다.
insertion_sort(A[1..N]) { # A[1..N]을 오름차순 정렬한다.
for i <- 2 to N {
loc = i - 1;
newItem = A[i];
# 이 지점에서 A[1..i-1]은 이미 정렬되어 있는 상태
while (1 <= loc and newItem < A[loc]) {
A[loc + 1] <- A[loc];
loc--;
}
if (loc + 1 != i) then A[loc + 1] = newItem;
}
}
입력
첫째 줄에 배열 A의 크기 N(5 ≤ N ≤ 10,000), 저장 횟수 K(1 ≤ K ≤ N2)가 주어진다.
다음 줄에 서로 다른 배열 A의 원소 A1, A2, ..., AN이 주어진다. (1 ≤ Ai ≤ 109)
출력
K 번째 저장 되는 수를 출력한다. 저장 횟수가 K 보다 작으면 -1을 출력한다.