[프로그래머스]더 맵게

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

코딩테스트 연습 - 더 맵게 | 프로그래머스

문제

• Leo는 모든 음식의 스코빌 지수가 K 이상이 될 때까지 반복하여 섞습니다. Leo가 가진 음식의 스코빌 지수를 담은 배열 scoville과 원하는 스코빌 지수 K가 주어질 때, 모든 음식의 스코빌 지수를 K이상으로 만들기 위해 섞어야 하는 최소 횟수를 return 하도록 solution 함수를 작성해라.

제한사항

• scoville의 길이는 1 이상 1,000,000 이하입니다.
• K는 0 이상 1,000,000,000 이하입니다.
• scoville의 원소는 각각 0 이상 1,000,000 이하입니다.
• 모든 음식의 스코빌 지수를 K 이상으로 만들 수 없는 경우에는 -1을 return 합니다.

입출력의 예

[scoville] - [1, 2, 3, 9, 10, 12]

    k       - 7

 return    - 2

 

입출력 예 설명

1. 스코빌 지수가 1인 음식과 2인 음식을 섞으면 음식의 스코빌 지수가 아래와 같이 됩니다.

   새로운 음식의 스코빌 지수 = 1 + (2 * 2) = 5

2. 스코빌 지수가 3인 음식 과 5인 음식을 섞으면 음식의 스코빌 지수가 아래와 같이 됩니다.

   새로운 음식의 스코빌 지수 = 3 * (5 * 2) = 13

   가진 음식의 스코빌 지수 = [13, 9, 10, 12]

모든 음식의 스코빌 지수가 7 이상이 되었고 이때 섞은 횟수는 2회 입니다.

 

풀이

 

우선 순위 큐를 정의하고 pq 변수에 스코빌 지수를 넣어준다.

* 우선순위 큐는 내부에서 오름차순을 해주기에 3, 2, 1 순으로 add 해주어도 poll()을 할 시 1의 값이 가장 먼저 나온다.

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

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

 

pq.poll()을 통해 가장 작은 값을 temPq1의 변수에 넣어 준다. 

첫 번째 조건은 스코빌 지수가 한개 이고, 그 값이 스코빌 지수 기준인 K 값에 미달 할 때 결과 값에 -1을 넣어준다.

두 번째 조건은 한개 뽑은 값이 스코빌 지수 기준 값에 미달일 때 가장 작은 스코빌 지수랑 그 다음 작은 스코빌 지수를

섞어 주는 조건이다.

세번 째는 조건은 모든 값이 스코빌 지수 기준 값이 상 일 때 while 값을 벗어 나기 위해 break 명령문을 사용하였다. 

   temPq1 = pq.poll();
			
	if ((pq.size() + 1) == 1 && temPq1 < K) {
		result = -1;
		break;
	} else if (temPq1 < K){
		result++;
		temPq2 = pq.poll();
		pq.add(temPq1 + (temPq2 * 2));
	} else {
		break;
	}

 

import java.util.PriorityQueue;
class Solution {
    public int solution(int[] scoville, int K) {
        int i = 0, result = 0, temp = 0, temPq1 = 0, temPq2 = 0;
        PriorityQueue<Integer> pq = new PriorityQueue<Integer>();
        
        for (int j = 0; j < scoville.length; j++) {
			pq.add(scoville[j]);
		}
        
		while (true) {
            
            temPq1 = pq.poll();
			
			if ((pq.size() + 1) == 1 && temPq1 < K) {
				result = -1;
				break;
			} else if (temPq1 < K){
				result++;
				temPq2 = pq.poll();
				pq.add(temPq1 + (temPq2 * 2));
			} else {
				break;
			} }

           }	
        return result;
    }
}