관리 메뉴

TEAM EDA

[프로그래머스] H-Index / 파이썬 본문

EDA Study/알고리즘

[프로그래머스] H-Index / 파이썬

김현우 2020. 4. 5. 18:31

문제 설명

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

입출력 예

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

입출력 예 설명

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

풀이

  • 기본적인 내용은 쉬웠지만 문제를 이해하기가 어려운 문제였습니다. 그리고 2가지 경우를 생각하기가 어려웠습니다.
    1. [22, 42]인 경우 결과값이 2인데 이 케이스
    2. [0] * 100000000000000000000으로 같은 값으로 반복되는 경우
  • 풀이는 min값부터 max값 까지 모두 순회하면서 H-Index를 찾고 최댓값을 찾는 방법입니다.
def solution(citations):
    # [22, 42] , 2인 경우 
    if min(citations) >= len(citations): answer = len(citations)
    else: answer = 0
    citations = sorted(citations)

    # [0] * 10000000 인 경우 
    for i in range(min(citations), max(citations) + 1):
        left = 0
        right = 0
        for j in citations:
            if j <= i: left += 1
            if j >= i: right += 1
        if (left <= i) & (right >= i):
            answer = max(answer, i)
    return answer

다른사람 풀이

  • 역순으로 뒤집어서 citations[i]와 i를 비교해서 작은 값을 선택
def solution(citations):
    citations.sort(reverse=True)
    answer = max(map(min, enumerate(citations, start=1)))
    return answer

출처: 프로그래머스 코딩 테스트 연습,https://programmers.co.kr/learn/challenges