호기심 많은 분석가

[프로그래머스] 정렬 - H-Index 본문

Coding/Coding Test & Algorithm

[프로그래머스] 정렬 - H-Index

DA Hun 2021. 6. 25. 16:10
 

코딩테스트 연습 - H-Index

H-Index는 과학자의 생산성과 영향력을 나타내는 지표입니다. 어느 과학자의 H-Index를 나타내는 값인 h를 구하려고 합니다. 위키백과1에 따르면, H-Index는 다음과 같이 구합니다. 어떤 과학자가 발표

programmers.co.kr

내 풀이

def solution(citations):
    for ind in range(len(citations)+1)[::-1] :
        if sum([1 if i>=ind else 0 for i in citations]) >= ind:
            return ind

 우선 문제의 내용이 너무 어려웠다. 테스트 케이스를 보고는 바로 이해하지 못했는데, 위키백과를 통해 완벽히 이해할 수 있었다. 1부터 하나씩 커가면서 n이상 인용된 논문이 n 편 이상인 최대 n을 구하라, n은 citations안에 없어도 된다.

 예를 들면 [25, 8, 5, 3, 3]의 경우 4이상인 값이 3개뿐이기 때문에 정답이 3. [10, 8, 5, 4, 3]의 경우는 정답이 4.

그래서 큰 index부터 표현해주기 위해 citations의 길이부터 하나씩 작아지면서 체크해줬다. 그 index보다 큰 값들의 개수가 index보다 클 경우 return ind를 통해 결과를 도출해내었다. 

다른 풀이

def solution(citations):
    citations.sort(reverse=True)
    answer = max(map(min, enumerate(citations, start=1)))
    return answer

 다른 사람들이 극찬하는 풀이법을 가져왔는데 꽤나 똑똑하다. 크기 순으로 나열해주고 enumerate와 start를 이용해 자신보다 큰 값들의 개수를 표현해준다. 그리고 그것이 역전되는 순간을 min과 max를 통해 구현해주었는데 enumerate를 잘 쓰는 것이 꽤나 코딩 테스트에서 도움이 되는 듯하다. 다음번에 적합한 문제가 있다면 시도해봐야겠다.

 하지만 이번 풀이는 난 내 풀이가 굉장히 직관적이고 마음에 든다.