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

프로그래머스

---

나의 풀이

• 각 h마다 추가되는 인용문 개수를 이분 탐색으로 찾아서 더했다.

from bisect import bisect_left, bisect_right

def solution(citations):
    citations.sort()
    # h ~ 0까지 가능
    cum_sum = dict(zip(range(len(citations)), [0] * len(citations)))
    # h보다 더 많이 인용된 논문 개수
    temp_sum = sum([1 for i in citations if i > len(citations)])
    for h in range(len(citations), -1, -1):
        right_idx = bisect_right(citations, h)     
        left_idx = bisect_left(citations, h)
        
        temp_sum += (right_idx - left_idx)
        # 최대 인용 순간
        if h <= temp_sum:
            break
            
    return h

 

다른 사람 풀이

• 인덱스 값과 같거나 큰 수의 개수를 [ic]l - i[/ic]설정하고 인용수와 비교했다.

def solution(citations):
    citations = sorted(citations)
    l = len(citations)
    for i in range(l):
        if citations[i] >= l-i:
            return l-i
    return 0

프로그래머스