프로그래머스 / level.2 / H-Index(C++)

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

문제 설명

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

내 해답

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int solution(vector<int> citations) {
    int answer = 0;
    
    // citations를 큰 수부터 정렬
    sort(citations.begin(), citations.end(),greater<int>());
    
    // 가장 많이 인용된 횟수부터 하나씩 줄여가며 조건에 맞는 h를 찾는다.
    int h = citations[0];   // 가장 많이 인용된 논문의 인용 횟수
    while(h > 0)
    {
        int above_h = 0;    // h번 이상 인용된 논문 수
        int below_h = 0;    // 나머지 논문 수
        
        // h번 이상 인용된 논문 수 찾기
        for(int j = 0; j < citations.size(); ++j)
        {
            if(h <= citations[j])
                ++above_h;
            else
                break;
        }
        
        // 나머지 논문 수 찾기
        below_h = citations.size() - above_h;
        
        // 조건 검사
        if(above_h >= h && below_h <= h)
            return h;
        
        // 조건에 부합하지 않을 시 h를 하나 줄여 탐색을 이어간다.
        else
            --h;
    }
    
    return answer;
}