[프로그래머스] Level2, H-Index (Python)

문제

https://programmers.co.kr/learn/courses/30/lessons/42747?language=python3 

코딩테스트 연습 - H-Index

 

코드

def solution(citations):
    for i in range(len(citations)):
        answer = 0
        for j in citations:
            if i <= j:
                answer += 1
        if i == answer:
            return i
    return answer

11번 케이스를 통과하지 못한다..

접근 방식이 잘못된 것 같다.

 

성공 코드

def solution(citations):
    answer = 0
    citations.sort()
    for i in range(1, len(citations)+1):
        num = citations[-i]
        if num >= i:
            answer = i
    return answer

정렬을 해준다음 맨 뒤에값 부터 i와 비교하면서 i보다 클 경우 answer에 넣어준다.

[0, 1, 3, 5, 6]이 있을 때 

 

h-index가 적합한 조건은 i번 이상 인용된 논문이 i개 이상 있어야한다.

정렬한 다음 맨 뒤에값부터 확인하는 이유는 i가 3일 경우 -3번째 인덱스가 i보다 클 경우에는 그 뒤에값은 모두 i모다 크므로 성립된다.

 

i = 1 시작하여 citations[-1] 은 6이므로 6이 1보다 크므로 answer은 1이 되고

i = 2 citations[-2]은 5이므로 2보다 크므로 answer은 2가 되고

i = 3 citations[-3]은 3이므로 3이랑 같으므로 answer은 3이된다.

i = 4 citations[-4]은 1이므로 4보다 작으므로 answer은 변하지 않는다.

i = 1,2,3이 h-index에 적합한데 그 중 최댓값을 구하므로 답은 3이된다.