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

Posted 2020-08-27 energy1010

我们可以通过二分查找法，在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;
}