返回列表中的第三个最小元素？ [复制]

Question

我正在尝试编写一个程序（在 python2.7 中），它接收整数列表并使用迭代返回列表中第三小的整数。如果有两个整数，则返回较大的整数。如果只有一个整数，则返回该整数。如果没有，则返回 none。 我很困惑，我不知道该怎么写。我在这里做了一个尝试： 更新版本：

s = raw_input("Enter your list:")
L = list(s)
sorted(L, key=int)
return [2]

if len(L) = [0:2]
return [1]

if len(L) = [0]
return 0

if len(L) = []
return none

但是，我在这方面遇到了一些困难。关于如何修改它以让它做我想做的任何建议？我知道堆函数，但我试图在不使用它的情况下编写它。非常感谢！

Answer 1

尝试使用：

def thirdSmallest(L):
    # sort the list from smallest to largest
    L.sort()
    # select the 3rd element, or if less than 3 elements exists,
    # select the last one of the sorted list.  If the list is empty return None
    return (L[:3] or [None])[-1]

Answer 2

ShadowRanger 是正确的。你应该使用heapq.nsmallest:

import heapq

def third_smallest(x):
  return (heapq.nsmallest(3, x) or (None,))[-1]

---

如果您正在寻找高性能选项，使用 heap 的 PriorityQueue 实现可能是最好的。

构建堆需要 O(n) 时间，然后只需弹出前三个元素 O(log n)。最终的时间复杂度是 O(n + log n) 与使用排序 O(n log n) 相比。

您的解决方案如下所示：

import heapq

def third_smallest(x):
  heapq.heapify(n)
  value = None
  try:
    for i in range(3):
      value = heapq.heappop(n)
  except IndexError:
    pass
  return value

这可能会得到改进。