★정렬★알고리즘 수업 - 선택 정렬 1[백준 파이썬 23881번]

23881 정렬

VERSION 1.0

import sys

def selection_sort(A ,k) : # A[1..N]을 오름차순 정렬한다.

    i = len(A)-1
    p = 0
    res = []
    while True:
        if i == 0:
            break
        if p>k:
            break
        A_idx = A.index(max(A[:i]))

        if max(A[:i]) > A[i]:
            idx = A[i]
            A[i] = max(A[:i])
            A[A_idx] = idx
            p+=1
            if p == k:
                res.append((idx,A[i]))
                res = res[0]
        else:
            i-=1

    if p<k:
        return print(-1)
    else:
        return print(*res)

N,K = list(map(int,sys.stdin.readline().split()))
A = list(map(int,sys.stdin.readline().split()))
selection_sort(A,K)

#%%

==> 쫌더 간단하게 추후에 풀어보자