[프로그래머스] 귤 고르기 / max_element, 내림차순 정렬

 

Level. 2

 

문제

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

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

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

* 제한사항
- 1 ≤ k ≤ tangerine의 길이 ≤ 100,000
- 1 ≤ tangerine의 원소 ≤ 10,000,000

 

풀이

#include <string>
#include <vector>
#include <algorithm>
#include <map>

using namespace std;

int solution(int k, vector<int> tangerine) {
    int answer = 0;
    map<int, int> map;
    for(int i=0; i<tangerine.size(); i++){
        map[tangerine[i]] += 1;
    }
    
    vector<int> v;
    for(auto m : map){
        v.push_back(m.second);
    }
    
    sort(v.begin(), v.end(), greater<>());
    int nowk = 0;
    for(int i=0; i<v.size(); i++){
        answer+=1;
        nowk += v[i];
        if(nowk >= k) break;
    }
    
    return answer;
}

 

 

 

---

다른 풀이

#include <bits/stdc++.h>

using namespace std;

int solution(int k, vector<int> tangerine) {
    int answer = 0;
    int m = *max_element(tangerine.begin(), tangerine.end());
    vector<int> v(m, 0);
    for(auto& t : tangerine){
        v[t - 1]++;
    }
    stable_sort(v.rbegin(), v.rend());
    for(int i = 0 ; i < v.size() ; i++){
        answer++;
        k -= v[i];
        if(k <= 0) return answer;
    }
    return answer;
}

 

 

 

1. max_element 

 

int m = *max_element(tangerine.begin(), tangerine.end());
vector<int> v(m, 0);

• max_element는 어떤 값을 가지는 배열, 벡터, 리스트 등이 있을 때 범위 내 최댓값을 반환해 주는 메서드이다.
• max_element를 사용하면 map을 사용하지 않고, 각각의 숫자가 몇 번 출현하는지를 저장할 수 있었다. 

 

2. 내림차순 정렬

 

// 오름차순 정렬
sort(v.begin(), v.end()); 

// 내림차순 정렬
reverse(v.begin(), v.end());	
sort(v.rbegin(), v.rend());		
sort(v.begin(), v.end(), greater<>());
stable_sort(v.rbegin(), v.rend());

 

 

rbegin, rend

 

• rbegin, rend는 역방향 반복자를 얻을 때 사용하는 함수이다.

 

stable_sort()

• 범위 내의 원소들을 오름 차순으로 안정 정렬을 수행한다 
• 안정 정렬 이란 크기가 같은 원소들의 위치가 바뀌지 않는 정렬이다. 

 

---

https://school.programmers.co.kr/learn/courses/30/lessons/138476

프로그래머스

http://www.soen.kr/lecture/ccpp/cpp4/39-3-3.htm

혼자 연구하는 C/C++ by WinApi