[프로그래머스] Lv1. 명예의 전당 (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       score                                                                                             result

3	[10, 100, 20, 150, 1, 100, 200]	[10, 10, 10, 20, 20, 100, 100]
4	[0, 300, 40, 300, 20, 70, 150, 50, 500, 1000]	[0, 0, 0, 0, 20, 40, 70, 70, 150, 300]

---

입출력 예 설명

입출력 예 #1

• 문제의 예시와 같습니다.

입출력 예 #2

• 아래와 같이, [0, 0, 0, 0, 20, 40, 70, 70, 150, 300]을 return합니다. 

 

 

 

나의 풀이

public int[] solution(int k, int[] score) {
   int[] answer = new int[score.length];
   ArrayList<Integer> arrayList = new ArrayList<>();
   
   int min = score[0];
   for (int i = 0; i< score.length; i++)
   {
      if (i < k) {
         arrayList.add(score[i]);
         min = Math.min(min, score[i]);
         answer[i] = min;
      }else {
         if (min < score[i]) {
            arrayList.add(score[i]);
            arrayList.remove(0);
            arrayList.sort(Integer::compareTo);
         }
         min = arrayList.get(0);
         answer[i] = arrayList.get(0);
      }
      arrayList.sort(Integer::compareTo);
      
   }
   
   return answer;
}

내가 푼 방식은 좀 복잡하게 ? 푼 것 같긴한데.. 일단 

명예의 전당 목록의 점수의 개수 k 까지는 arraylist에 담아주고, 그 중 최소값을 구해서 answer 배열에 담아준다. 

그리고 k개 이상으로 넘어가면 이제 score의 값이 min보다 크면 arraylist를 정렬해서 가장 작은 값은 제거하고, score 값을 넣어준다. 

뭔가 코드 더 정리할 수 있을 것 같긴해서 나중에 수정해보는 것도 좋을 것 같다. 이건 그냥 답만 나오게한 코드같은 기분...😮

 

 

 

다른 사람 풀이

public int[] otherSolution(int k, int[] score) {
   int[] answer = new int[score.length];
   
   PriorityQueue<Integer> priorityQueue = new PriorityQueue<>();
   
   for (int i = 0; i< score.length; i++)
   {
      priorityQueue.add(score[i]);
      
      if (priorityQueue.size() > k) {
         priorityQueue.poll();
      }
      answer[i] = priorityQueue.peek();
   }
   
   return answer;
}

아.. 자료 구조 더 공부해야겠다고 생각이 드는 코드다..

priorityQueue를 사용하면 이렇게 쉽게 해결되는데... 나는 엄청 복잡하게 만든 것이다..ㅠㅠㅠㅠㅠ

반성하게 되는구나.. 자바 공부를 소홀히 하니까 이러는구나... 오늘이라도 알았으니까 빨리 정리해서 이제 이걸 사용하도록 하면 되니까 ㅎㅎㅎ 괜찮아!! 

까먹지 않게 오늘 TIL 주제로 priorityQueue를 공부해야겠다..!!