프로그래머스 - H-Index / Python

문제: https://programmers.co.kr/learn/courses/30/lessons/42747

코딩테스트 연습 - 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회 이하입니다.

입출력 예

citationsreturn

[3, 0, 6, 1, 5]

3

입출력 예 설명

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

 

---

정답

def solution(citations):  # 논문 별 인용 횟수
    answer = 0

    citations.sort(reverse=True)

    for idx, c in enumerate(citations):
        if c < idx + 1:
            break
        else:
            answer = idx + 1

    return answer

 

---

풀이

def solution(citations):  # 논문 별 인용 횟수
    answer = 0

    citations.sort(reverse=True)

    for idx, c in enumerate(citations):
        if c < idx + 1:
            break
        else:
            answer = idx + 1

    return answer

 h번 이상 인용된 논문이 h 편 이상일 때, h의 최댓값을 구하는 문제입니다. 문제가 복잡할 수 있으나 이를 1부터 시작해서 보면 쉽습니다. 입출력 예를 기준으로 하겠습니다.

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

 

1. 1편 이상 인용된 논문이 1편 이상인가? 

 3, 6, 1, 5 총 4개이므로 맞습니다.

 

2. 2편 이상 인용된 논문이 2편 이상인가? 

 3, 6, 5 총 3개이므로 맞습니다.

 

3. 3편 이상 인용된 논문이 3편 이상인가? 

 3, 6, 5 총 3개이므로 맞습니다.

 

4. 4편 이상 인용된 논문이 4편 이상인가? 

 6, 5 총 2개이므로 틀립니다.

 

5. 5편 이상 인용된 논문이 5편 이상인가? 

 6, 5, 총 2개이므로 틀립니다.

 

 이를 통해 h의 최댓값은 3인 것을 알 수 있습니다. 

 

이 문제는 정렬을 통해 풀 수 있습니다. 먼저 citations를 역순으로 정렬합니다. 

 

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

 

 그리고 해당 논문이 인용된 수와 1부터 시작하는 인덱스와 비교하여 인덱스가 더 큰 값이 나올 때까지 반복합니다. 

 

1 < 6 : answer = 1

2 < 5 : answer = 2

3 < 3 : answer = 3

4 > 1 : break

 

 이렇게 역순으로 조회하면 H-Index를 효율적인 방법으로 구할 수 있습니다.