[프로그래머스] Lv2 - H-Index

 

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

 

입출력 예

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

 

해답 및 해설

파이썬 (Python)

if citations[i]<=i+1:
    answer = citations[i]
    break

처음에는 단순하게 입출력 예시만 보고, 정렬된 리스트에서 다음과 같이 하면 되지 않을까 생각했었다. 하지만 이는 좀 더 다양한 테스트 케이스를 생각하지 못 한 거였다.

 

[1, 1, 1, .... ]

[0]

[1000]

[999, 998, 997]

이외 테스트 케이스 예시

 

다음과 같은 극단적인 테스트 케이스들이 있을 수 있었고, 이것들을 코너 케이스라 부른다고 한다. 이후 이런 코너 케이스에도 통할 수 있게 코드를 바꾸었다.

 

def solution(citations):
    citations.sort(reverse=True)
    n = len(citations)
    
    max_h = 0
    for i in range(len(citations)):
        h = min(i+1, citations[i])
        if h > max_h:
            max_h = h

    return max_h

원소의 정렬을 한 후 순서와 인용수 중 작은 값을 h라 잡고 이 h가 가장 크게 되는 경우를 골라주었다.

 

def solution(citations):
    citations.sort(reverse=True)
    answer = max(map(min, enumerate(citations, start=1)))
    return answer

다른 사람의 풀이를 보니 이걸 map과 enumerate 함수를 이용하여 한줄로 푼 풀이가 존재했다.