找轮转后的有序数组中第K小的数

时间:2024-01-09 20:08:02

我们可以通过二分查找法,在log(n)的时间内找到最小数的在数组中的位置,然后通过偏移来快速定位任意第K个数。
此处假设数组中没有相同的数,原排列顺序是递增排列。
在轮转后的有序数组中查找最小数的算法如下:

int findIndexOfMin(int num[],int n) {
int l = 0;
int r = n-1;
while(l <= r) {
int mid = l + (r - l) / 2;
if (num[mid] > num[r]) {
l = mid + 1;
} else {
r = mid - 1;
}
}
return l;
}

接着基于此结果进行偏移,再基于数组长度对偏移后的值取模,就可以找到第K个数在数组中的位置了:

int findKthElement(int num[], int m, int k)
{
if (k > m) return -1;
int base = findIndexOfMin(num, 0, m-1);
int index = (base+k-1) % m;
return index;
}