[LG전자] 코테준비 05. "H-Index"프로그래머스 python

H-Index

• 링크: https://school.programmers.co.kr/learn/courses/30/lessons/42747
• 난이도: level 2
• 분류: 정렬
• 직접 푼 여부: yes
• 언어: 파이썬

 

 

정답 코드 💻✔

사실 이문제는 범위가 크지 않아서, 그냥 n다 돌지만, 만약에 n이 너무 크면, 이분탐색해도 좋을 것 같다.

아래 코드서 stopped 의 이유는, [7, 6, 5] 와 [5, 4, 1]의 경우 둘다 i 가 2인상태에서 끝나지만, 앞은 3 뒤는 2가 답인 경우이므로, stopped라는 플레그를 만들어 break에 걸려서 나온 것과 아예 끝난 것을 구분해 주었다. 

def solution(citations):
    citations.sort(reverse=True)
    stopped=False
    for i in range(len(citations)):
        if citations[i]<(i+1):
            stopped = True
            break
    return i if stopped else i+1