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.
题目要求:给定一个升序的citations数组,设计算法,在时间复杂度为O(log n)的情况下完成H-inde的查找。
思路:使用二分查找的思路。设定两个指针left=1,right=n,分别指向paper数的左右两端。每次检查之间值mid是否符合条件。
Java代码:
/**
* 使用二分查找的思路
*/
public class Solution {
public int hIndex(int[] citations) {
int n = citations.length;
int index = 0;
int left = 1;
int right = n;
while (left <= right) {
//找到中间的篇数
int mid = (left + right) / 2;
//如果从后面数第mid篇的引用数大于等于mid
if (citations[n - mid] >= mid) {
//再看第mid篇前面的paper的引用数是否小于等于mid,如果是就找到一个
if ((n - mid - 1 < 0) || (citations[n - mid - 1] <= mid)) {
index = mid;
left += 2;//让mid增大1
} else {//否则,该mid不符合规则,存在更大的解,mid需要增大
left = mid + 1;
}
} else {//否则,说明mid后面不够mid篇的引用数大于mid,需要将mid减小。
right = mid - 1;
}
}
return index;
}
}