정렬 - H-Index

아래 문제는 프로그래머스에서 제공하는 코딩테스트 > 정렬 > H-Index (고득점 Kit)의 문제입니다🧑🏻‍💻

 

문제 제시

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회 이하입니다.

입출력 예

[3, 0, 6, 1, 5]

3

입출력 예 설명

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

 

문제 해결

import Foundation

func solution(_ citations:[Int]) -> Int {
    let stack: [Int] = citations.sorted(by: >)
    
    for i in 0..<stack.count {
        if i >= stack[i] {
            return i
        }
    }
    return stack.count
}

사용된 개념

 - 정렬

 - 배열

 - 조건/반복문

 

문제 뒷담화

이번 문제는 짧지만 잘 이해해야된다.

[100, 90]의 입력값이 주어지는 경우도 생각을 해야한다. 이 경우에는 논문이 2편이고 인용이 다 2를 넘으니까 H-Index가 2이다.

처음 내림차순으로 정렬을 시켜준다.

그 후 배열의 수 만큼 반복을 돌며 배열의 인덱스와 값을 비교해준다. 인덱스가 0부터 시작함으로 인덱스가 값을 넘어가는 순간이

최대 H-Index값이 된다. 그럴 경우 해당 인덱스 정수를 반환해준다.

만약 조건을 성립하지 않는다면 최대 H-Index는 배열에 담긴 값의 수임으로 stack.count를 반환해준다.

되게 생각해보면 단순한 문제인데 막상 코드로만 짜려면 생각이 잘 나지 않았다.

그래서 스도코드처럼 직접 손으로 테스트 케이스를 그려보며 생각했다.

어차피 내림차순이니까 해당 되는 인덱스 번호가 있을것이고 H-Index보다 적어도 들어있는 값이 더 클거라 생각하고 값과 인덱스의 상관관계로 접근했다.

다른 사람의 풀이중 더 효율성이 좋은 코드를 찾았다.

조건을 돌때 정렬을 시키고 enumerated()를 사용하여 인덱스와 값을 저장할 수 있고 바로 비교할 수 있다.

import Foundation

func solution(_ citations:[Int]) -> Int {
    for (index, cit) in citations.sorted(by: >).enumerated() {
        if index >= cit {
            return index
        }
    }
    return citations.count
}

 

 

[참고자료]

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