[프로그래머스] H-Index / 코틀린

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

코딩테스트 연습 - H-Index

 

문제

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

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

어떤 과학자가 발표한 논문의 인용 횟수를 담은 배열 citations가 매개변수로 주어질 때, 이 과학자의 H-Index를 return 하도록 solution 함수를 작성해주세요.

제한사항

• 과학자가 발표한 논문의 수는 1편 이상 1,000편 이하입니다.
• 논문별 인용 횟수는 0회 이상 10,000회 이하입니다.

입출력 예

citations	return
[3, 0, 6, 1, 5]	3

입출력 예 설명

이 과학자가 발표한 논문의 수는 5편이고, 그중 3편의 논문은 3회 이상 인용되었습니다. 그리고 나머지 2편의 논문은 3회 이하 인용되었기 때문에 이 과학자의 H-Index는 3입니다.

※ 공지 - 2019년 2월 28일 테스트 케이스가 추가되었습니다.

---

 

의식의 흐름

1. 어려운 문제가 아니라고 느꼈음에도 수십번 다시 품. 그 이유는?

2. H-Index가 citations 안에 있는 값이라고 생각했기 때문이다.

3. H-Index는 논문 인용 수와 전혀 상관 없다.. ㅠㅠㅠ ㅎㅏ

4. h값은 항상 n을 넘을 수 없다는 점만 잘 생각하면 됐던 거였다.

 

해결 코드

fun solution(citations: IntArray): Int {

    var arr = citations.sortedByDescending { it }
    var answer = arr.size // // h는 항상 <= n


    arr.forEach {
        var count = arr.filter { e -> e >= answer }.count()

        if (count >= answer && arr.size - count <= answer) {
            return answer
        } else {
            answer--
        }
    }
    return answer
}

n값을 초기로 부여 후, 하나씩 줄여가면서 찾다가 걸리면 그게 max값이라 리턴했다.

반례는 [0,0,0,0,0] (답은 0) 이나  [5,5,5,5] (답은 4)로 테스트 해보면 된다.