使用递归在Python列表中查找Kth最大元素

时间:2016-06-29 02:35:37

标签: python recursion

给定一个包含一些随机未排序数字的输入列表,我试图编写一个程序,输出该列表中第k个最大的不同元素。例如:

Input: 
el = [10,10, 20,30,40, 40]
k = 2
Output: 30 #Since 30 is the second largest distinct element in the list

以下函数将输入列表,数据透视索引和k作为输入,并填充列表" lessser" ,所有元素都小于数据透视表并填充另一个列表 "更大" 所有元素都大于数据透视。

现在,查看两个列表的长度,我可以确定第k个最大元素是在较小列表中还是在较大列表中。现在我递归调用相同的函数。但是,对于某些k值,我的程序输出是错误的。

def kthLargest(el, pivotIndex, k):
    pivot = el[pivotIndex]
    lesser = [] #List to store all elements lesser than pivot
    greater = [] #Lsit to store all elements greater than pivot
    equals = [] #List to store all elements equal to pivot

    for x in el:
        if x > pivot:
            greater.append(x)
        elif x < pivot:
            lesser.append(x)
        else:
            equals.append(x)
    g = len(greater) #Length of greater list
    l = len(lesser) 

    if(g == k - 1): #If greater list has k-1 elements, that makes the pivot kth largest element
        return pivot
    elif(g < k):
        return kthLargest(lesser, l - 1, k) #If greater list is smaller than k, kth largest element is in lesser list
    else:
        return kthLargest(greater,  g - 1, k) #Else kth largest element is in greater list

6 个答案:

答案 0 :(得分:1)

你有什么理由要使用递归吗?要查找列表的第k个最大元素,您必须查看整个列表,因此问题本质上是O(n)复杂性。

你可以这样做而不需要这样的递归:

el = [10, 10, 53, 20, 30, 40, 59, 40]
k = 2

def kth_largest(input_list, k):
    # initialize the top_k list to first k elements and sort descending
    top_k = input_list[0:k]
    top_k.sort(reverse = True)

    for i in input_list[k:]:
        if i > top_k[-1]:
            top_k.pop() # remove the lowest of the top k elements
            top_k.append(i) # add the new element
            top_k.sort(reverse = True) # re-sort the list

    return top_k[-1] # return the kth largest


kth_largest(el, k)

答案 1 :(得分:1)

使用递归可以轻松解决此问题。我只是不确定为什么你需要问题描述中的支点...例如:

def find_kth(k, arr):
    if k == 1:
        return max(arr)
    m = max(arr)
    new_arr = list(filter(lambda a: a != m, arr))
    return(find_kth(k-1, new_arr))

答案 2 :(得分:1)

这是一个简单的解决方案:

def kthmax(k, list):
    if (k == 1):
        return max(list)
    else:
        m = max(list)
        return(kthmax(k-1, [x for x in list if x != m]))

kthmax(3,[4, 6, 2, 7, 3, 2, 6, 6])

输出:4

答案 3 :(得分:0)

如果我们可以传入已经按降序排序的列表或系列 订单,例如

el.sort_values(ascending=False, inplace = True) 

然后你可以很容易地找到第k个最大(索引,值)元组 只使用排序的数据帧列和/或列表的简单切片

def kth_largest(input_series, k):
    new_series = input_series[k-1:len(input_series)]
    return (np.argmax(new_series) , np.max(new_series))

el = pd.Series([10, 10, 53, 20, 30, 40, 59, 40])
print el
k = 2

el.sort_values(ascending=False, inplace=True)

print kth_largest(el, 2)

Output: 30
0    10
1    10
2    53
3    20
4    30
5    40
6    59
7    40
dtype: int64
(2, 53)

答案 4 :(得分:0)

算法:取最大值的索引并转换为零。

def high(arr,n):
    for i in range(n+ 1 ):
        arr[arr.index(max(arr))] = 0
        return max(arr)
high([1,2,3,4,5], 2)

答案 5 :(得分:0)

我找到第K个最大元素的方法是...

lst=[6,2,3,4,1,5]
print(sorted(lst,reverse=True)[k-1])