已知n长度的元素序列,列出所有栈输出的可能结果
今天上数据结构课时遇到了一个题目:
设数据元素序列{a,b,c,d,e,f,g}的进堆栈操作和出堆栈操作可任意进行(排除堆栈为空时的出栈情况),下列哪些元素序列可由堆栈序列得到?
哇…这要一个一个列出来岂不是疯掉…
因为对公式还不是很了解,所以决定先把代码写出来,把所有可能都列出来,这样说不定会好理解一点
其实用递归做起来还是蛮简单的,栈的输出结构画张图可以发现和二叉树结构类似
这样,代码就比较容易实现了,一开始打算用C语言来实现的,结果太久没碰C语言,能算出多少种,但输出结果不会打出来…所以又改用Python来实现
def copyData(tmpStack,tmpRes,popTime,pushTime):
# copy
return tmpStack[:],tmpRes[:],popTime,pushTime
def printAllRes(tmpStack,tmpRes,popTime,pushTime,count):
# 参数:临时栈,显示结果,出栈次数,入栈次数,数据总长度
if(len(tmpRes) == count):
# 当出栈次数等于数据总长度,将临时显示的结果添加到总结果中
results.append(tmpRes)
if(pushTime < count):
# 当入栈次数小于上限,优先进行入栈操作
t_tmpStack,t_tmpRes,t_popTime,t_pushTime = copyData(tmpStack,tmpRes,popTime,pushTime)
t_tmpStack.append(data[t_pushTime])
printAllRes(t_tmpStack,t_tmpRes,t_popTime,t_pushTime+1,count)
if(popTime < count and len(tmpStack) > 0):
# 当出栈次数小于上限并且临时存储的栈不为空时,出栈
t_tmpStack,t_tmpRes,t_popTime,t_pushTime = copyData(tmpStack,tmpRes,popTime,pushTime)
t_tmpRes.append(t_tmpStack.pop())
printAllRes(t_tmpStack,t_tmpRes,t_popTime+1,t_pushTime,count)
if __name__ == "__main__":
data = [1,2,3] # 待入栈的数据
results = [] # 临时存放所有结果的列表
printAllRes([],[],0,0,len(data))
print('共有' + str(len(results)) + '种方式')
for item in results:
print(item)
刚开始因为不太常用递归,没有加入 t_ 的临时变量…,怎么做都不对,改了无数次都不对,一个晚上的时间都花在这上面了,甚至都开始怀疑python不能递归了😂,看来还是自己不熟练啊…