프로그래머스 H-Index - 1

 

def solution(citations):
    n = len(citations)
    h = n
    while (h > 0):
        if sum(citations) >= n * h:
            return h
        h -= 1
    return 0

https://school.programmers.co.kr/learn/courses/30/lessons/42747

프로그래머스

 

어떤 과학자가 발표한 논문 n편 중, h번 이상 인용된 논문이 h편 이상이고 나머지 논문이 h번 이하 인용되었다면 h의 최댓값이 이 과학자의 H-Index입니다.

 

즉 h번 이상 인용된 논문이 h편 이상, h번 이하 인용된 논문이 (n-h)편 이하라는 말인데

이걸 수식으로 좀 정리해보면, 어떤 리스트 A가 주어졌을 때

h * h + (n-h) * h <= sum(A) 라고 할 수 있다.

정리하면 n * h <= sum(A)가 된다.

 

그래서 이 수식을 만족하는 h의 최댓값을 찾으면 된다 생각했다.

최댓값을 구하는 거는 반복문을 이용하면 되니까

def solution(citations):
    n = len(citations)
    h = n
    while (h > 0):
        if sum(citations) >= n * h:
            return h
        h -= 1
    return 0

 

이렇게 작성했다.

하지만 통과하질 못했다.

우선 질문하기 기능에서 사용할 수 있는 반례는 찾았다.

[1, 1, 2, 2, 3, 4, 5, 6, 7]

 

해당 반례를 내 코드로 돌리면 h는 3이 나오지만 실제 h는 4이다.

이를 해결할 수 있는 방법을 생각해봐야겠다.