[코드트리] TreeSet 연습문제

문제 : top K 숫자

n개의 숫자가 주어졌을 때, 중복을 제외하고 내림차순으로 정렬했을 때 앞에 있는 k개의 숫자를 출력하는 프로그램을 작성해보세요.

---

입력 형식

첫 번째 줄에는 원소의 개수 n과 k가 공백을 사이에 두고 주어집니다.

두 번째 줄에는 n개의 원소가 공백을 사이에 두고 주어집니다.

• 1 ≤ k ≤ n ≤ 100,000
• 1 ≤ 주어지는 원소 값 ≤ 109

출력 형식

중복을 제외하고 내림차순으로 정렬했을 때 앞에 있는 k개의 숫자를 공백을 사이에 두고 출력합니다. 중복을 제외했을 때 원소의 개수가 k보다 작은 경우는 없다고 가정해도 좋습니다.

 

 

나의 풀이

 

#include <iostream>
#include <set>
using namespace std;

int main() {
    int n; int k; int a;
    set<int> s;
    cin >> n >> k;
    for(int i=0; i<n; ++i){
        cin >> a;
        s.insert(a);
    }
    for(int i=0; i<k; ++i){
        cout << *s.rbegin() << " ";
        s.erase(*s.rbegin());
    }
    return 0;
}

 

마지막 원소 출력 후, 마지막 원소를 삭제하는 것을 반복하는 식으로 풀어서 해결했는데

해설에는 다른 방식으로 설명이 되어있었음.

대충 입력값에 -1을 곱해서 set에 집어넣고,,, (내림차순을 위해서)

아래 반복문을 사용해서 앞에서부터 이터레이터를 사용해서 출력함.

for(set<int>::iterator it = s.begin(); cnt < k; it++)

 

이런 방법도 있구만~~~

 

근데 아직은 iterator 사용에 익숙하지 않아서 ,,, 공부 더 해야겠다.

코드트리는 UI가 신생 사이트 같아서 별 기대를 하지 않았는데 예상외로 문제 퀄리티가 좋은듯!!

 

 

문제 출처 코드트리