[프로그래머스] K번째 수
April 20, 2021
https://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 함수를 작성해주세요.
풀이
문제를 보고 이분탐색이다 싶었다. h-index 를 찾아야 할 수로 두고 각 mid 값이 H-index 의 조건에 만족하는지 확인해주어서 이 조건을 만족하는 최대값을 구했다.
코드
{% raw %}
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
bool isValid(int target, vector<int> citations) {
int idx;
for (idx = 0; idx < citations.size(); idx++) {
if (citations[idx] >= target) break;
}
return (citations.size() - idx >= target);
}
int binarySearch(int max, vector<int> citations) {
int head = 0;
int tail = max;
int maxH = 0;
while (head <= tail) {
int mid = (head + tail) / 2;
bool valid = isValid(mid, citations);
if (valid) {
maxH = mid;
head = mid + 1;
}
else {
tail = mid - 1;
}
}
return maxH;
}
int solution(vector<int> citations) {
int answer = 0;
sort(citations.begin(), citations.end());
return binarySearch(citations.size(), citations);
}
{% endraw %}
