题目描述(中等难度)
求 H-Index
,和 上一道题 一样,只不过这道题给定的数组是有序的,详细的可以先做一下上一题。
解法一
先看一下之前的其中一个的做法。
我们从 0
开始遍历,依次判断点是否在直线下方,如果出现了点在直线上方(包括在直线上),那么当前点的垂线与直线交点的纵坐标就是 H-Index
了。
点的垂线与直线交点的纵坐标的求法是 n - i
,n
是数组长度,i
是数组下标。
代码如下。
public int hIndex(int[] citations) { Arrays.sort(citations); int n = citations.length; for (int i = 0; i < n; i++) { // 点在直线上方 if (citations[i] >= n - i) { return n - i; } } return 0; }
说白了,我们是要寻找第一个在直线上方(包括在直线上)的点,给定数组是有序的,所以我们可以用二分查找。
public int hIndex(int[] citations) { int n = citations.length; int low = 0; int high = n - 1; while (low <= high) { int mid = (low + high) >>> 1; //在直线上方 if (citations[mid] >= n - mid) { if (mid == 0) { return n; } //前一个点是否在直线下方 int before = mid - 1; if (citations[before] < n - before) { return n - mid; } high = mid - 1; } else { low = mid + 1; } } return 0; }
总
主要就是二分查找的应用了。