매운 것을 좋아하는 Leo는 모든 음식의 스코빌 지수를 K 이상으로 만들고 싶습니다. 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 Leo는 스코빌 지수가 가장 낮은 두 개의 음식을 아래와 같이 특별한 방법으로 섞어 새로운 음식을 만듭니다.
섞은 음식의 스코빌 지수 = 가장 맵지 않은 음식의 스코빌 지수 + (두 번째로 맵지 않은 음식의 스코비 지수 * 2)
Leo는 모든 음식의 스코빌 지수가 K 이상이 될 때까지 반복하여 섞습니다. Leo가 가진 음식의 스코빌 지수를 담은 배열 scoville과 원하는 스코빌 지수 K가 주어질 때, 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해서 섞어야 하는 최소 횟수를 return 하도록 solution 함수를 작성해주세요.
ex) scoville = [1, 2, 3, 9, 10, 12] and K = 7 일 때,
- 스코빌 지수가 1인 음식과 2인 음식을 섞으면 음식의 스코빌 지수가 아래와 같이 됩니다.
새로운 음식의 스코빌 지수 = 1 + (2 * 2) = 5
가진 음식의 스코빌 지수 = [5, 3, 9, 10, 12] - 스코빌 지수가 3인 음식과 5인 음식을 섞으면 음식의 스코빌 지수가 아래와 같이 됩니다.
새로운 음식의 스코빌 지수 = 3 + (5 * 2) = 13
가진 음식의 스코빌 지수 = [13, 9, 10, 12]
모든 음식의 스코빌 지수가 7 이상이 되었고, 이때 섞은 횟수는 2회 입니다.
접근 point
- Priority Queue (오름차순 정렬)
Step 1. 오름차순으로 정렬되는 Priority Queue 생성하여 scoville 지수 값을 삽입한다.
Step 2. Priority Queue 맨 위의 값이 K보다 작거나 같을 때 까지 아래와 같이 반복한다.
Step 3. Priority Queue의 사이즈가 2보다 작고, 맨 위의 값이 K 보다 작을 때는 -1을 return 한다.
** 더 이상 스코빌 지수를 계산 할 음식 두 가지가 없는 것을 의미
#include <bits/stdc++.h>
using namespace std;
priority_queue<int, vector<int>, greater<int>> pq; // 오름차순
int solution(vector<int> scoville, int K) {
int cnt = 0;
for(int i = 0; i < scoville.size(); i++) {
pq.push(scoville[i]);
}
while(true) {
if(pq.size() >= 2) {
if(pq.top() < K) {
int a = pq.top(); pq.pop();
int b = pq.top(); pq.pop();
pq.push(a + 2 * b);
cnt++;
} else {
break;
}
} else {
if(!pq.empty()) {
if(pq.top() < K) cnt = -1;
}
break;
}
}
return cnt;
}
'프로그래머스' 카테고리의 다른 글
프로그래머스 lvl 2. 영어 끝말잇기 [C++] (0) | 2023.07.07 |
---|---|
프로그래머스 lvl 2. 귤 고르기 [C++] (0) | 2023.07.05 |
프로그래머스 lvl 2. 전화번호 목록 [C++] (0) | 2023.07.03 |
프로그래머스 lvl 2. 숫자의 표현 [C++] (0) | 2023.07.02 |
프로그래머스 lvl 2. 구명보트 [C++] (0) | 2023.06.16 |