使用递归返回嵌套列表中的第二个最小数字

时间:2017-02-26 05:03:34

标签: python algorithm list recursion

我必须使用递归返回python列表中的第二小数字,并且无循环。我所做的是创建一个辅助函数,它返回列表中(最小,第二小)值的元组,然后我只需tuple[1]函数中的second_smallest

def s_smallest(L):

    if(len(L) == 2):
        if (L[0] >= L[1]):
            return (L[1],L[0])
        else:
            return (L[0],L[1])
    else:
        first_smallest,second_smallest = s_smallest(L[1:])

        if L[0] >= first_smallest and L[0] <= second_smallest:
            return (first_smallest, L[0])

        elif L[0] <= first_smallest:
            return (L[0], first_smallest)

        else:
            return (first_smallest, second_smallest)

这有效,但现在我需要处理嵌套列表,因此s_smallest([1,2,[3,0]])应该返回(0,1)。我试过这样做:

if isinstance(L[0],list):
    first_smallest,second_smallest = s_smallest(L[0])
else:
    first_smallest,second_smallest = s_smallest(L[1:])

如果是列表,则获取第一个最小值和第二个最小值,但是我收到错误builtins.TypeError: unorderable types: int() >= list()。如何解决此问题以处理嵌套列表?

2 个答案:

答案 0 :(得分:5)

我可能会建议将已删除的列表和最小值减少分成两个单独的,明确定义的任务

  • deepReduce将使用指定的缩减功能缩小列表列表
  • deepMin使用deepReduce
  • 执行min
import math # used for math.inf

def min (x,y):
  return x if x < y else y

def deepReduce (f, y, xs):
  if not xs:
    return y
  elif isinstance(xs[0], list):
    return deepReduce(f, deepReduce(f, y, xs[0]), xs[1:])
  else:
    return deepReduce(f, f(y, xs[0]), xs[1:])

def deepMin (xs):
  return deepReduce (min, math.inf, xs)


data = [1,2,[7,[6,1,3,[0,4,3]],3,4],2,1]
print(deepMin(data))
# 0

哦,但是你说你想要第二个最小的数字。让我们做一些代码的返工。当然我一直都知道,但两次回答这个问题可以让我展示这个具体实现的多功能性 - 粗体的变化

def min2 (xs, y):
  # x1 is the smallest, x2 is second smallest
  x1, x2 = xs
  if (y < x1) and (y < x2):
    return (y, x2)
  elif y < x2:
    return (x1, y)
  else:
    return (x1, x2)

def deepMin2 (xs):
  # notice we change to use tuple of math.inf now
  x1, x2 = deepReduce (min2, (math.inf, math.inf), xs)
  return x2

data = [1,2,[7,[6,1,3,[0,4,3]],3,4],2,1]
print(deepMin2(data))
# 1

我应该指出,我们根本不需要触摸deepReduce,这就是重点 - 我们应该可以在嵌套列表上执行任意深度操作,而无需静态编码行为进入我们的职能。

现在,您可以编写所需的深度减速器,并使用deepReduce

进行调用

答案 1 :(得分:0)

完整解决方案

除了functools.reduce之外什么也没用,没有循环来处理任意嵌套列表:

import functools

def helper(acc, x):
    if type(x) is list:
        return functools.reduce(lambda acc, x: helper(acc, x), x, acc)
    else:
        if x < acc[0]:
            return (x, acc[0])
        elif x < acc[1]:
            return (acc[0], x)
        else:
            return (acc[0], acc[1])

def second_smallest(l):
    if len(l) < 2:
        return None
    else:
        if l[0] <= l[1]:
            return functools.reduce(lambda acc, x: helper(acc, x), l[2:], (l[0], l[1]))
        else:
            return functools.reduce(lambda acc, x: helper(acc, x), l[2:], (l[1], l[0]))

>>> second_smallest([1,2,[0,3,[-1,-2]]])
(-2, -1)