[Swift]프로그래머스_H-Index

문제 설명

 

H-Index는 과학자의 생산성과 영향력을 나타내는 지표입니다. 어느 과학자의 H-Index를 나타내는 값인 h를 구하려고 합니다. 위키백과에 따르면, 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

 

 

풀이

내가 푼 방법

 

내가 푼 것은 for문을 이용해서 직접 비교하는 방법을 이용했다. 고민없이 쉽게 풀었지만 생각해보면 조금만 배열이 커져도 시간이 오래 걸릴 것이다.

func solution(_ citations:[Int]) -> Int {

    var possible: [Int] = []

    for h in 0...citations.count {
        if citations.filter {$0 >= h}.count >= h {
            possible.append(h)
        }
    }
    return possible.max()!
}

단순하게 작은 수부터 조건을 만족하는지 확인하고 만족하는 h값을 배열에 넣어 최댓값을 리턴하게끔했다.

 

 

다른 분이 푸신 방법

 

정렬 후에 인덱스를 이용한 방법이다. 처음에 어떤 의미인지 이해하는데 시간이 조금 걸렸다. 하지만 이해하고 난 뒤 굉장히 쉽고 간단한 방법이라는 것을 알게됐다. 

위의 예시를 들어 설명해보겠다. [3, 0, 6, 1, 5]를 정렬하면 [6, 5, 3, 1, 0]일 것이다. 인덱스는 0부터 시작하므로 h의 값을 인덱스 + 1이라고 생각해보자. 차례대로 for문을 돌면된다. 

먼저, h가 1일 때(i가 0일때) citations[i]의 값은 6이다. 즉, 1번 이상 인용된 논문이 일단 하나 이상있다는 뜻이다. 마찬가지로 h가 2일 때는 citations[i]의 값은 5이다. 즉, 2번 이상 인용된 논문이 2편 이상(6번 인용된 논문, 5번 인용된 논문)있다는 것을 의미한다.

이런식으로 진행하다가 h의 값이 4일 때 citations[i]의 값은 1이다. 이는 더 이상 h의 값이 커질 수 없다는 것을 의미한다(내림차순 정렬을 했기때문에 점점 작은 숫자가 나오게 되므로). 따라서 이때의 h값은 조건을 만족하지 못하므로 h값이 아닌 1 작은 인덱스(i)의 값을 리턴해주면된다.

만약, 배열의 원소들이 배열의 count(논문의 편수)보다 전부 클 경우엔 count로 바로 리턴하도록 해준다.

func solution(_ citations:[Int]) -> Int {
    
    var result: Int = 0
    let sorted = citations.sorted{$0 > $1}

    if sorted.min()! > sorted.count {
        result = sorted.count
    } else {
        for i in 0..<sorted.count {
            if i+1 > sorted[i] {
                result = i
                break
            }
        }
    }
    return result
}

 

결론

 

아무리 쉽게 풀었어도 더 효율적으로 풀 수 있는지 계속 확인해봐야겠다. 이번 문제같은 경우도 내 방법과 다른 분의 방법에서 시간 차이가 많이 났기 때문에 시간복잡도를 생각하면서 코드를 짜도록 노력해야겠다.