[C++][프로그래머스] 귤 고르기

프로그래머스

 

📄 문제

경화는 과수원에서 귤을 수확했습니다. 경화는 수확한 귤 중 'k'개를 골라 상자 하나에 담아 판매하려고 합니다. 그런데 수확한 귤의 크기가 일정하지 않아 보기에 좋지 않다고 생각한 경화는 귤을 크기별로 분류했을 때 서로 다른 종류의 수를 최소화하고 싶습니다.

예를 들어, 경화가 수확한 귤 8개의 크기가 [1, 3, 2, 5, 4, 5, 2, 3] 이라고 합시다. 경화가 귤 6개를 판매하고 싶다면, 크기가 1, 4인 귤을 제외한 여섯 개의 귤을 상자에 담으면, 귤의 크기의 종류가 2, 3, 5로 총 3가지가 되며 이때가 서로 다른 종류가 최소일 때입니다.

경화가 한 상자에 담으려는 귤의 개수 k와 귤의 크기를 담은 배열 tangerine이 매개변수로 주어집니다. 경화가 귤 k개를 고를 때 크기가 서로 다른 종류의 수의 최솟값을 return 하도록 solution 함수를 작성해주세요.

 

📝 풀이

#include <string>
#include <vector>
#include <map>
#include <queue>

using namespace std;

int solution(int k, vector<int> tangerine) {
	map<int, int> m;

	for (int n : tangerine)
		++m[n];

	// 우선순위 큐를 이용하여 남은 귤의 개수가 적은것부터 내림차순
	priority_queue<pair<int, int>> q;
	for (pair<int, int> p : m)
		q.push({ -p.second, p.first });

	// left: 제거해야 할 귤의 개수
	int left = tangerine.size() - k;
	for (int i = 0; i < tangerine.size() - k; ++i)
	{
		// 현재 제거할 귤의 개수
		int cnt = -q.top().first;

		// left가 적어 제거하지 못하는경우 리턴
		if (cnt > left)
			break;
		// 제거 가능한경우에는 큐에서 pop
		else
		{
			left -= cnt;
			q.pop();
		}
	}

	return q.size();
}

• map과 우선순위 큐를 이용하여 문제를 해결합니다.
• 귤을 사이즈별로 map에 저장하여 개수를 갱신합니다.
• 우선순위 큐를 이용하여 귤의 개수가 적은것부터 내림차순하여 정렬한 후 제거해야 할 귤의 개수만큼 큐에서 제거하여 마지막으로 큐의 개수를 리턴합니다.