20230615 TIL

코딩테스트 레벨 1 명예의 전당(1)

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

입출력 예시

내가 작성한 코드

k 일까지는 도전한 참가자들의 score가 다 명예의 전당에 올라간다. 그렇기 때문에 k일까지만큼의 score를 잘라서 명예의 전당 배열에 넣어두고 result에는 명예의 전당 배열에서 가장 작은 값을 k만큼 넣어둔다.
k 일 후에는 명예의 전당 배열에 있는 수 중 가장 작은 수와 비교해서 큰 값만 명예의 전당 배열에 넣어두고 가장 작은 값을 result에 넣어두었다.

function solution(k, score) {
    var answer = [];
    let klist = [] 
    if (score.length < k){
        k = score.length
    }
   //k일 전
    for(let i =0;i<k;i++){
        klist.push(score[i])
        answer.push(klist.sort((x,y) => y-x)[klist.length-1])
    }
    //k일 후
    for (let i=k;i<score.length;i++){
        const list = klist.sort((x,y) => y-x).slice(0,k)
        if(list[k-1] < score[i]){
            klist.push(score[i])
        }
        const answerlist = klist.sort((x,y) => y-x).slice(0,k)
        answer.push(answerlist[k-1])
    }
    return answer; 
}

다른사람이 작성한 코드

다른 사람들이 작성한 코드를 보다가 reduce를 사용해서 작성한 코드를 봤는데 더 간단하고 깔끔하게 작성되어 있었다. 나는 reduce는 생각도 못했는데 신기하다. 
동작 방식은 내 코드와 비슷한것 같다. stack이 명예의 전당 배열 역할을 하는 것 같고 k 일 전까지는 다 넣어서 올림차순으로 정렬한다. k 일 후에는 올림차순으로 정렬한 후 맨 앞의 수(가장 작은 수)를 뺀다. 이렇게 만든 배열에서 가장 앞의 값을 a로 return 해준다.

function solution(k, score) {
    const stack = []
    return score.reduce((a,c) => {
        if(stack.length < k) {
            stack.push(c)
            stack.sort((a,b) => a - b)
        }
        else {
            stack.push(c)
            stack.sort((a,b) => a - b)
            stack.shift()
        }
        a.push(stack[0])
        return a
    },[])
}