Follow up for H-Index: What if the citations
array is sorted in ascending order? Could you optimize your algorithm?
Hint:
Expected runtime complexity is in O(log n) and the input is sorted.二分法。
C++:
// Runtime: 12 ms
class Solution {
public:
int hIndex(vector& citations) {
int len = citations.size();
int left = 0;
int right = len - 1;
while (left <= right)
{
int mid = left + (right - left) / 2;
if (citations[mid] >= len - mid)
{
right = mid - 1;
}
else
{
left = mid + 1;
}
}
return len -left;
}
};
Java:
// Runtime: 12 ms
public class Solution {
public int hIndex(int[] citations) {
int len = citations.length;
int left = 0;
int right = len - 1;
while (left <= right) {
int mid = left + (right - left) / 2;
if (citations[mid] >= len - mid) {
right = mid - 1;
}
else {
left = mid + 1;
}
}
return len - left;
}
}