[프로그래머스 / python] H-Index

프로그래머스

문제 설계

오름차순 정렬하고 이분탐색이용해서 , h를 0부터 논문리스트에서 최대 인용 횟수까지 반복문
1. h번 이상 인용된 논문 수 구할 때는 bi_left로 구하고 리스트길이 - 그 리턴 인덱스가 h번 이상 인용된 논문 수
2. h번 이하 인용된 논문 수 구하는 건 bi_right로 구하고 리스트 길이 - 그 리턴 인덱스가 h번 이하 인용된 논문 수
1과 2 조건을 둘 다 만족하는 건 결과 리스트에 넣어준다.
그리고 최대 값 리턴

 

문제 풀이

from bisect import bisect_left,bisect_right
def solution(citations):
    
    answer = []
    n=len(citations)
    citations.sort()
    h=max(citations)
    
    for i in range(0,h+1):
        up=n-bisect_left(citations,i)
        down=n-bisect_right(citations,i)
        if up>=i and down<=i:
            answer.append(i)

    return max(answer)

정리

nums=[0,1,2,3,4,5] 일 때

bisect_left(nums,2) 는 리턴값이 2(인덱스)

bisect_right(nums,2)는 리턴값이 3(인덱스)