programmers | Lv2. H-Index [Python]

💡 Python 3

🔗 문제

H-Index [Link]

💻 코드

효율성 제로 내 코드

def solution(citations):
    answer = 0
    citations.sort()
    for i in range(max(citations)+1):
        count = 0
        for c in citations:
            if i <= c: count += 1
        if count >= i and len(citations)-count <= i: answer = max(answer, i)
    return answer

💥 다른 사람 코드

대박 !! 완전 깔끔하면서 내장 함수를 잘 활용한 코드

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

• sort로 가장 큰 값부터 작은 값으로 정렬

• enumerate로 (index, value)형태로 묶음

• 최댓값(start = 1)부터 각 value에 대해 최솟값 value의 값을 min으로 추출

  • 예를들어 [6, 5, 4, 1, 0]의 경우에선 min~ 부분은 min(1, 6), min(2, 5), min(3, 4), min(4, 1), min(5, 0), 즉 해당 인용수 이상의 논문개수와 해당 논문의 인용수 중 더 작은 숫자를 고르는 작업을 함

• 이 추출된 값은 enumerate가 끝나는 citations 리스트의 크기에 해당하는 갯수 반환

• map으로 묶으면 한 value의 입장에서 보는 최솟값 value의 집합, 즉 h값들의 집합 반환

• h값중 최대값을 max로 뽑아서 출력