[프로그래머스] 명예의 전당 (1)

문제

"명예의 전당"이라는 TV 프로그램에서는 매일 1명의 가수가 노래를 부르고, 시청자들의 문자 투표수로 가수에게 점수를 부여합니다. 매일 출연한 가수의 점수가 지금까지 출연 가수들의 점수 중 상위 k번째 이내이면 해당 가수의 점수를 명예의 전당이라는 목록에 올려 기념합니다. 즉 프로그램 시작 이후 초기에 k일까지는 모든 출연 가수의 점수가 명예의 전당에 오르게 됩니다. k일 다음부터는 출연 가수의 점수가 기존의 명예의 전당 목록의 k번째 순위의 가수 점수보다 더 높으면, 출연 가수의 점수가 명예의 전당에 오르게 되고 기존의 k번째 순위의 점수는 명예의 전당에서 내려오게 됩니다.

이 프로그램에서는 매일 "명예의 전당"의 최하위 점수를 발표합니다. 예를 들어, k = 3이고, 7일 동안 진행된 가수의 점수가 [10, 100, 20, 150, 1, 100, 200]이라면, 명예의 전당에서 발표된 점수는 아래의 그림과 같이 [10, 10, 10, 20, 20, 100, 100]입니다.

명예의 전당 목록의 점수의 개수 k, 1일부터 마지막 날까지 출연한 가수들의 점수인 score가 주어졌을 때, 매일 발표된 명예의 전당의 최하위 점수를 return하는 solution 함수를 완성해주세요.

제한사항

3 ≤ k ≤ 100

7 ≤ score의 길이 ≤ 1,000

0 ≤ score[i] ≤ 2,000

---

풀이

일단, 내가 이해한 문제는

k 명의 가수가 명예의 전당에 오를 수 있다.

함수의 실행 결과는, 매일매일의 명예의 전당에 오른 가수의 최저점을 요소로 가진 배열이 되어야 한다.

이 두 가지의 조건을 만족해야 하는 것이 포인트인 문제이다.

그러면 일단 길이가 k인 배열을 생성하고, 여기에 명예의 전당 가수들의 점수를 넣으면 되겠다고 생각했다.

k개의 점수를 모두 저장하고, 오름차순으로 정렬하고 제일 낮은 점수와 새로이 명예의 전당 등록을 기다리는 가수의 점수를 비교하는 작업을 한다.

기존 명예의 전당에 등록된 제일 낮은 점수를 삭제하고, 그 자리에 새로 명예의 전당에 들어갈 점수를 삽입하는 작업을 반복하면? 문제를 풀 수 있을 것 같다!

아래는 내가 처음 작성한 코드이다.

func solution(_ k:Int, _ score:[Int]) -> [Int] {
    var rank: [Int] = [], result: [Int] = []
    for i in 0..<k {
        rank.append(score[i])
        rank = rank.sorted()
        result.append(rank[0])
    }
    for i in k..<score.count {
        rank[0] = rank[0] < score[i] ? score[i] : rank[0]
        rank = rank.sorted()
        result.append(rank[0])
    }
    return result
}

2개의 반복문을 사용해서 문제를 해결하려 했다. 무조건 처음부터 k번 째 점수를 받는 가수는 명예의 전당에 올라가기 때문에(앞에 아무도 없으니) k번 째 가수들의 점수를 rank 배열에 저장하고 이를 정렬했다. 이어지는 두 번째 반복문에서는 rank의 0번 인덱스 값(제일 작은 값)과 새로운 가수의 점수를 비교해서 명예의 전당에 등록한다. 이 모든 과정 중에 제일 작은 명예의 전당 점수를 result 배열에 저장하면 되는데...

반복문 두 개를 나누어 사용해놓고, 이 둘이 같은 인덱스를 사용한다고 착각했었다. 바보인가?

저런 식으로 문제를 풀어내니, 전체 가수들보다 명예의 전당 인원 수가 더 많을 때는 오류가 발생되었다.

문제를 해결하기 위해서 반복문을 하나만 사용하기로 마음먹고 아래와 같이 다시 코드를 작성했다.

import Foundation

func solution(_ k:Int, _ score:[Int]) -> [Int] {
    var rank: [Int] = [], result: [Int] = []
    for i in 0..<score.count {
        if i < k { rank.insert(score[i], at: 0) }
        else { if rank[0] < score[i] { rank.remove(at: 0)
            rank.insert(score[i], at: 0)
        } }
        rank.sort()
        result.append(rank[0])
    }
    return result
}

입력으로 받은 score 배열의 길이만큼 반복하는 구문에서, 명예의 전당에 들어갈 k번 째 가수들 점수를 rank 배열에 저장하는 작업을 하고 정렬한 후에 제일 작은 값을 result 배열에 더한다. k명의 가수를 지난 후에는, 제일 작은 점수와 새로 들어올 점수를 비교해서 뺄건 빼고, 넣을건 넣어서 재정렬하는 작업을 반복... 하다보면 결과가 나온다.

print(solution(3, [10, 100, 20, 150, 1, 100, 200]))

결과 : [10, 10, 10, 20, 20, 100, 100]
위와 같이 정확한 결과가 나오는 것을 보고 마무리!

---

회고

.remove(at: ) 메소드와, .insert(_ newElement: , at: )메소드를 잘 활용하면, 한정된 요소만을 가지는 배열을 유지하는데 많은 도움이 될 듯 하다는 생각을 했다. 특히 insert 메소드는 원하는 위치에 원하는 값을 넣을 수 있다는 특징이 마음에 든다. 앞으로 배열을 사용하는 알고리즘 문제를 풀어낼 때 자주 사용할 것 같다.