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; } }