[프로그래머스 lv2] 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회 이하입니다.

입출력 예

입출력 예

입출력 예 설명

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

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

---

나의 풀이

def solution(citations):
    citations.sort(reverse=True)
    for i in range(len(citations)):
        if i+1 > citations[i]:
            return i
    return len(citations)

이런 문제에서는 주어진 리스트를 정렬하는 것이 편하다. (여기서는 내림차순 정렬)

두 가지 경우

citations로 [3, 0 , 6, 1, 5] 가 주어진 경우와 citations로 [8, 10, 10, 15, 17, 22, 24, 28, 32, 42, 47] 이 주어진 경우를 동시에 살펴봤다.

내림차순으로 정렬 후 최초로 i + 1 보다 citations[i+1] 의 값이 작아진 경우를 찾는다.

• 위의 경우에서는 i+1이 4일 때 citations[i+1]인 1보다 처음으로 작아졌기 때문에 i 값인 3이 h가 된다.
• 아래의 경우는 i+1이 11일 때 citations[i+1]인 8보다 처음으로 작아졌기 때문에 i 값인 10이 h가 된다.

만약 반복문이 끝나고도 i+1이 citations[i+1]보다 작은 경우가 없었다면 논문의 개수(citations의 개수)만큼이 h가 된다.

 

풀이에 대한 이유는 다음과 같다.

i 의 의미에 대해서 잘 생각해보아야 한다.

내림차순으로 정렬했기 때문에 i + 1은 '개수'의 의미를 가질 수 있다.

즉 citations[i]가 i+1보다 크거나 같다면 인용된 횟수가 i+1번인 논문의 개수가 i개라는 뜻이다.

i를 하나씩 증가시키면서 최초로 이러한 규칙이 깨지는 부분이 생기면 그 때의 i 값이 최대 h 값이 되는 것이다.

 

테스트 결과