[프로그래머스] (LV2) H-Index

H-Index

JavaScript, Python

문제 설명

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입니다.

※ 공지 - 2019년 2월 28일 테스트 케이스가 추가되었습니다.

---

1. https://en.wikipedia.org/wiki/H-index “위키백과” ↩

풀이

---

javascript

function solution(citations) {
    var answer = 0;
    var sorted = citations.sort((a,b)=>b-a)

    for(let i = 0; i < sorted.length; i++){
        if(answer > sorted[i])
            return answer;
        else
            answer = (sorted[i] <= (i+1) ? sorted[i] : i+1)
    }
    return answer
}

python

def solution(citations):
    h = 0
    for i, c in enumerate(sorted(citations,reverse=True)):
        if h > c:
            return h
        else:
            h = (c if c <= (i+1) else i+1)
    return h

설명

---

정렬(sort)하지 않고서는 O(N²)이상의 풀이 방법이 없었다.

배열을 내림정렬을 한 뒤, 앞에서 부터 index와 값을 비교하여 최대값을 갱신시켜주면 된다.

여기서 index는 자연스럽게 현재 값보다 같거나 큰 값들의 개수가 된다(정확히는 i+1이).

내가 염려했던 부분은 [5,5,5,5,0]와 같은 배열이었는데, 이 배열의 h값은 4가 나와야 한다.

즉, 배열에 포함되어 있지 않은 값으로도 h가 갱신되어야 한다.

따라서 index와 값을 비교하여 더 낮은 값으로 갱신을 진행했다.

정렬 한 뒤에는 O(N)풀이 인데, 평균적으로는 훨씬 적을 것으로 예상된다.