프로그래머스 | Lv.2 H-index (파이썬)

문제 설명

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

입출력 예

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

입출력 예 설명

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

 

풀이

(1) 나의 풀이 - 오름차순 정렬 이용

n-i는 h-index를 최댓값으로 찾아야하므로 가장 최댓값인 n(len(citations))에서 0부터 줄여나가는 것, 

그 값이 현재 논문의 인용된 횟수와 처음으로 같거나 혹은 횟수보다 작아지는 순간에 리턴합니다.

def solution(citations):
    n = len(citations)
    citations.sort()
    for i in range(n):
            if n-i <= citations[i]:
                return n-i
    return 0

(2) 다른 풀이 - 내림차순 정렬 이용

이 풀이가 좀 더 잘 읽히는 듯 하다.

인덱스와 내림차순한 citation을 enumerate로 튜플로 만든다.

인덱스와 citation값을 비교하여 인덱스가 크거나 같아지는 순간의 인덱스를 리턴한다.

위 오름차순 풀이와 반대로 푸는 방법이다.

다만 if문을 만족하지 않을때는 len(citation)을 리턴해줘야한다.

예)

[5,10,15,20,25] 

[3,3,3]

= 인덱스보다 항상 citation이 크거나 같다

def solution(citations):
    citations.sort(reverse=True)
    for idx , citation in enumerate(citations):
        if idx >= citation:
            return idx
    return len(citations)