23882: 알고리즘 수업 - 선택 정렬 2

문제

오늘도 서준이는 선택 정렬 수업 조교를 하고 있다. 아빠가 수업한 내용을 학생들이 잘 이해했는지 문제를 통해서 확인해보자.

N개의 서로 다른 양의 정수가 저장된 배열 A가 있다. 선택 정렬로 배열 A를 오름차순 정렬할 경우 K 번 교환이 발생한 직후의 배열 A를 출력해 보자.

크기가 N인 배열에 대한 선택 정렬 의사 코드는 다음과 같다.

selection_sort(A[1..N]) { # A[1..N]을 오름차순 정렬한다.
    for last <- N downto 2 {
        A[1..last]중 가장 큰 수 A[i]를 찾는다
        if (last != i) then A[last] <-> A[i]  # last와 i가 서로 다르면 A[last]와 A[i]를 교환
    }
}

입력

첫째 줄에 배열 A의 크기 N(5 ≤ N ≤ 10,000), 교환 횟수 K(1 ≤ K ≤ N)가 주어진다.

다음 줄에 서로 다른 배열 A의 원소 A1, A2, ..., AN이 주어진다. (1 ≤ Ai ≤ 109)

출력

 K 번 교환이 발생한 직후의 배열 A를 한 줄에 출력한다. 교환 횟수가 K 보다 작으면 -1을 출력한다.

예제 입력 1 복사

5 2
3 1 2 5 4

예제 출력 1 복사

2 1 3 4 5

3 1 2 5 4 (4와 5가 교환됨) -> 3 1 2 4 5 (A[1..4]에서 4가 가장 크므로 교환이 발생하지 않음) -> 3 1 2 4 5 (2와 3이 교환됨) -> 2 1 3 4 5 (1과 2가 교환됨) -> 1 2 3 4 5. 총 3회 교환이 발생하고 두 번째 교환 직후 배열 A의 모습은 2 1 3 4 5이다.

예제 입력 2 복사

5 4
3 1 2 5 4

예제 출력 2 복사

-1

교환 횟수 3이 K 보다 작으므로 -1을 출력한다.

N,K = map(int, input().split())
AList = list(map(int, input().split()))

# 선택정렬은 현재 인덱스와 다음인덱스들을 비교하여, 교환합니다.
#이 문제에서는 뒤에서부터 인덱스를 비교합니다.
cnt = 0 #교환횟수
for i in range(N-1,-1,-1): #몇 바퀴 도는 지
    if(K<cnt):
        break
    maxA =i;
    for j in range(i-1,-1,-1): #i번 인덱스와 그 다음 인덱스들 비교
        if AList[maxA] < AList[j]:
            maxA = j     #최소값 저장
    if(i!=maxA):
        cnt+=1
        AList[i] ,AList[maxA]= AList[maxA], AList[i] 
    if(cnt==K):
        print(' '.join(map(str, AList)))
        break
if(K>cnt):
    print(-1)

 

PyPy3로 제출하니 시간초과가 나지 않았습니다.

문제 이해가 까다로운 문제였습니다.