[programmers] 더 맵게

문제
https://school.programmers.co.kr/learn/courses/30/lessons/42626

프로그래머스

 

---

< 더 맵게 >

 

문제 풀이 (Java)

import java.util.*;

class Solution {
	public int solution(int[] scoville, int K) {
		int answer = 0;

		PriorityQueue<Integer> queue = new PriorityQueue<>();

		for (int i = 0; i < scoville.length; i++) {
			queue.add(scoville[i]);
		}

		while (!queue.isEmpty() && queue.peek() < K) {
			if (queue.size() == 1) {
				answer = -1;
				break;
			}

			int num1 = queue.poll();
			int num2 = queue.poll();
			queue.add(num1 + (num2 * 2));
			answer += 1;
		}
		return answer;
	}
}

 

우선순위 큐에 scoville 배열에 있는 값들을 다 저장한다. queue의 peek 값이 K보다 작을 때 다음 과정을 반복한다.

 

1. queue의 크기가 1이라면 모든 음식의 스코빌 지수를 K 이상으로 만들 수 없으므로 answer에 -1을 저장하고 종료

2. queue에서 값 2개 pop

3. 가장 맵지 않은 음식의 스코빌 지수 + (두 번째로 맵지 않은 음식의 스코빌 지수 * 2)를 queue에 저장

4. answer +1

 

최종 answer을 반환한다.

---

---

 

출처: 프로그래머스 코딩 테스트 연습, 
https://school.programmers.co.kr/learn/challenges