algorithm - 在单峰数组中找到第 k 个元素

Question

给定一个包含 n 个不同元素的单峰数组 A（意味着它的条目按升序排列，直到其最大元素，之后其元素按降序排列），一个整数 p（即增加的第一部分的长度）和 k （第 k 个最小元素）给出一个算法来计算在 O(log n) 时间内运行的第 k 个最小元素的值。

例子：

A= {1,23,50,30,20,2} 
p= 2
k=3

答案：20

编辑

我试过这个：

def ksmallest(arr1, arr2, k):
if len(arr1) == 0:
    return arr2[len(arr2)-k-1]
elif len(arr2) == 0:
    return arr1[k]
mida1 = (int)(len(arr1)/2)
mida2 = (int)((len(arr2)-1)/2)
if mida1+mida2<k:
    if arr1[mida1]>arr2[mida2]:
        return ksmallest(arr1, arr2[:mida2], k-(len(arr2)-mida2))
    else:
        return ksmallest(arr1[mida1+1:], arr2, k-mida1-1)
else:
    if arr1[mida1]>arr2[mida2]:
        return ksmallest(arr1[:mida1], arr2, k)
    else:
        return ksmallest(arr1, arr2[mida2+1:], k)

Answer 1

对于初学者来说，再看看你的索引。你开始：

if len(arr1) == 0:
    return arr2[len(arr2)-k-1]
elif len(arr2) == 0:
    return arr1[len(arr1)-k-1]

但是如果 arr1 是升序，arr2 是降序，那么肯定不会在同一位置找到第 k 个最小元素。