[Programmers] H-Index (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입니다.

문제 링크
H-Index Wiki

풀이 ✏️

처음에 문제를 잘못읽어서 1개의 테스트 케이스 빼고 모두 실패를 했고, H-Index의 wiki를 읽어보고 정확하게 이해를 했다. 답이 citations의 원소 중에 있어야 하는 것은 아니라는 점 이다..
제출하고 2-3개의 테스트 케이스가 실패가 되어서 다양한 케이스를 찾느라 시간이 조금 걸렸던 문제이다.
예외 케이스만 정확하게 판단하면 짧은 코드로 풀이할 수 있다.

코드(python)

1. h-index가 될 수 있는 0~citations의 길이를 돌면서 체크한다.
2. h-index 이상의 값을 가지는 원소를 index라는 배열에 담고, 배열의 길이도 h-index 이상인지를 체크한다.
3. 두 가지를 만족하지 않을 때까지 체크하여 최대값을 구한다.
4. citations의 길이(l)보다 크거나 같은 값이 l개 이상일 경우 반복문 이후에 처리를 해준다.

def solution(citations):
    answer = 0
    citations.sort()
    for i in range(len(citations)):
        index = [citation for citation in citations if citation >= answer]
        if len(index) >= answer:
            answer += 1
        else:
            return answer - 1

    index = [citation for citation in citations if citation >= answer]
    return len(index)

이렇게 풀어도 정답으로 처리는 되었지만, 다른 분들의 풀이를 참고하니 아래와 같이 훨씬 간단하게도 가능했다.

1. citations를 역순으로 정렬한다.
2. 반복문 내의 i (= h-index의 의미)가 가능한지 체크하고 가능하면 바로 리턴
3. 반복문을 끝까지 돈 경우, 모든 원소가 h-index이상이므로 길이를 리턴

def solution2(citations):
    citations = sorted(citations, reverse=True)
    for i in range(len(citations)):
        if citations[i] <= i:
            return i
    return len(citations)

github 코드

테스트 케이스

아래 테스트 케이스를 참고하였다.

citations1 = [3, 0, 6, 1, 5] #ans:3
citations2 = [4, 4, 4] #ans:3
citations3 = [22, 32] #ans:2
citations4 = [0, 0, 1, 1] #ans:1
citations5 = [6, 6, 6, 6, 6, 1] #ans:5