[c++] 프로그래머스 :: H-index (문제 풀이, 코드)

H-index

문제

https://programmers.co.kr/learn/courses/30/lessons/42747#

문제 설명

H-Index는 과학자의 생산성과 영향력을 나타내는 지표입니다. 어느 과학자의 H-Index를 나타내는 값인 h를 구하려고 합니다. 위키백과1에 따르면, H-Index는 다음과 같이 구합니다.

어떤 과학자가 발표한 논문 n편 중, h번 이상 인용된 논문이 h편 이상이고 나머지 논문이 h번 이하 인용되었다면 h의 최댓값이 이 과학자의 H-Index입니다.

어떤 과학자가 발표한 논문의 인용 횟수를 담은 배열 citations가 매개변수로 주어질 때, 이 과학자의 H-Index를 return 하도록 solution 함수를 작성해주세요.

제한사항

• 과학자가 발표한 논문의 수는 1편 이상 1,000편 이하입니다.
• 논문별 인용 횟수는 0회 이상 10,000회 이하입니다.

입출력 예

citations	return
[3, 0, 6, 1, 5]	3

입출력 예 설명

이 과학자가 발표한 논문의 수는 5편이고, 그중 3편의 논문은 3회 이상 인용되었습니다. 그리고 나머지 2편의 논문은 3회 이하 인용되었기 때문에 이 과학자의 H-Index는 3입니다.

 

코드

#include <string>
#include <vector>
#include <set>

using namespace std;

int solution(vector<int> citations) {
    int answer = 0;
    int size = citations.size();
    set<int> result;

    for(int i=1; i<=10000; i++){ // 10^4
        int tmp = 0;
        int flag = 0;
        for(int j=0; j<size; j++){ // 10^3
            if(citations[j]>=i)
                tmp++;
            if(tmp>=i){
                flag =1;
                break;
            }
        }
        if(flag==1){
            result.insert(i);
        }
    }
    answer = *(result.end()--);
    return answer;
}

 

풀이

우선 처음에는 단순하게 주어진 배열을 정렬하여, 뒤에서부터 요소들을 체크하면서 해당 요소의 값이 h-index에 해당하는지를 검사하는 식의 알고리즘을 짜봤다. 하지만 곧 정답 값이 배열 내의 숫자에서만 나오는 것이 아니라는 것을 깨달았다. 예를 들어 [4,4,4]라는 입력값에는 3이라는 정답이 나올 수가 있다.

따라서 단순하게 수를 1씩 증가시키면서 체크해봐야겠다고 생각했고, 입력 범위가 크지 않아서 가능할 것이라고 판단하고 알고리즘을 짜봣다.

• 1부터 1씩 증가시키면서 수를 따진다.
• 해당 수보다 큰 수가 배열에 해당 수만큼 있다면 h-index가 될 수 있는 후보이다.
• 따라서 배열을 훑으면서 대소 관계를 따진다.
• 후보가 되는 수들은 모두 set에 넣어서 마지막에 가장 큰 수를 정답으로 둔다.

 

채점 결과

 

다른 사람의 풀이

 이진 탐색으로 푼 사람도 있었다. 주어진 입력의 범위가 0에서 10,000까지이니, 이를 이진탐색해서 푼 방식이다. 이진 탐색을 생각했으면서 왜 이 풀이로 향하지 못했는지... 흠.. 다음 번에는 더 생각해서 알고리즘을 짜봐야겠다. 시간복잡도를 10^7 에서 10^4 정도로 줄일 수 있다.