stack = [1,2,3] 逆序之后就是 stack = [3,2,1]
怎么实现这个功能呢?
我们先写一个reverse(stack),返回逆序好的栈
def reverse(stack):
do something
...
return stack
容易发现,当栈为空时不需要返回,可以把栈为空作边界条件
reverse函数又丰富了一些
def reverse(stack):
if not stack:
return
else:
do something
...
return stack
只能用递归函数实现,递归:把原问题转化为规模更小的问题。这里面主要考虑,1递归函数要不要返回,返回什么 2 是由子问题的递归函数得到原问题的解,还是把原问题分解成子问题去解决。 如果是通过子问题的解来得到原问题的解,就像是归并排序的递归解法一样,可以假设 子问题已经解决了,可以直接用子问题的结论, 能这样理解很重要。 如果是把原问题分解成子问题来解决, 那么实现细节就会是 把一个复杂问题分解成 几个子问题,比如把 A(复杂问题,需要递归解决的) 分解成 a,b,c三个子问题,再分别解决a,b,c。和前一种递归情况不同的是,不能假设子问题已经解决了, 不能用子问题的解来获取原问题的解,这种情况其实就是所说的分治,快速排序的递归解法就属于这种情况。
从stack中取出一个元素后, 设 stack 变为了 newstack
考虑 reverse(stack) 和 reverse(newstack) 的联系。
两个都是返回一个逆序的栈, 原问题返回的栈比子问题多一个元素。
stack = [1,2,3] 那么 reverse(stack)=[3,2,1]
如果newstack = [2,3] 那么 reverse(newstack)=[3,2]
原问题就可以转为为 在子问题返回的栈底加一个 1,栈不能直接在栈底添加元素,所以这种递归思路行不通。
如果 newstack = [1,2],原来的栈取出的元素是3
reverse(newstack)=[2,1], 再添加一个3就是reverse(stack)的解了,这种递归思路是可行的。剩下的问题就是怎么从stack=[1,2,3],取出栈底的元素,可以先假设由getlast(stack)这个函数完成, 那么主函数reverse(stack)就完整了
def reverse(stack):
if not stack:
return
# 取出栈底元素
last = getlast(stack)
# 把取出栈底元素的栈 逆序
reverse(stack)
# 上一步已经把栈逆序了,已经实现了,完成了。只要把原来栈底元素再加上,整个栈就逆序了。
stack.append(last)
return stack
下面考虑怎么实现getlast(stack),取出栈底的元素,注意是只取出栈底的元素, stack=[1,2,3] 取出栈底的元素之后是 [1,2]
getlast(stack)的功能是取出栈底元素, 那么 getlast([1,2,3]) 和getlast([2,3])的返回时一样的,即原问题的解就是子问题的解,假设子问题已经完成,直接使用子问题的返回
def getlast(stack):
# 容易发现当栈只有一个元素时,可以直接取出栈底元素,栈长度为1可以作为边界条件
if len(stack) == 1:
return stack.pop()
else:
# 栈弹出一个元素,问题规模会减小,最终达到边界条件
peek = stack.pop()
# 直接使用子问题的解
result = getlast(stack)
# 把栈顶的元素放回去,不然栈的元素就弹出完了
stack.append(peek)
return result
下面是完整实现
def getlast(stack):
if len(stack) == 1:
return stack.pop()
else:
peek = stack.pop()
result = getlast(stack)
stack.append(peek)
return result
def reverse(stack):
if not stack:
return
last = getlast(stack)
reverse(stack)
stack.append(last)
return stack