[프로그래머스] H-Index (Lv.2) - 파이썬(Python)

문제풀이

 H-Index는 과학자의 생산성과 영향력을 나타내는 지표입니다. 어느 과학자의 H-Index를 나타내는 값인 h를 구하려고 합니다. 위키백과에 따르면, 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
[1, 2, 3, 5, 6, 7, 10, 11]	5
[1, 11, 111, 1111]	3

 

입출력 예 설명

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

---

문제 풀이

이 문제를 한 줄로 정리하면, 

한 과학자의 논문 중에서, 적어도 h번 이상 인용된 논문이 h편 이상인 경우의 최대값을 구하는 문제다.

 

• 과학자가 발표한 논문 중에서 h번 이상 인용된 논문이 h편 이상이고,
• 나머지 논문들이 h번 이하로 인용되었다면, h가 이 과학자의 H-Index입니다.

예제, [6, 5, 3, 1, 0] 를 보면,

 

• 첫 번째 논문은 6번 인용됨: 1 <= 6 → 참
• 두 번째 논문은 5번 인용됨: 2 <= 5 → 참
• 세 번째 논문은 3번 인용됨: 3 <= 3 → 참
• 네 번째 논문은 1번 인용됨: 4 <= 1 → 거짓
• 다섯 번째 논문은 0번 인용됨: 5 <= 0 → 거짓

네 번째 논문이 1번 인용되었고, h값을 4로 가정했을 때, 위 조건에 값을 대입하여 비교해보겠습니다.

• h(1)번 이상 인용된 논문이 h(4)편 이상 있어야 합니다.

네 번째 논문이 1번 인용되었기 때문에, 이 논문이 h번 이상 인용되려면 h ≤ 1이어야 합니다. 그러나 h를 4로 가정했기 때문에 조건을 만족하지 않습니다.

def solution(citations):
    answer = 0
    
    # 1. 내림차순으로 정렬
    cit = sorted(citations, reverse = True) 
    
    # 2. 인덱스와 인용 횟수를 순차적으로 확인
    for num, citations in enumerate(cit):
        # 3. 인용 횟수가 인덱스보다 크다면
        # 코드상 인덱스가 0부터 시작하기 때문에 +1을 해줌
        if num+1 <= citations:
            # 4. H-Index 갱신
            answer = num+1
    
    return answer