前面提到樹的3中遍歷迭代實現的時候，講到：還有一種思路，直接針對遞歸的實現方式，想辦法通過棧來實現遞歸，得到的還是和上面一樣

一般的遞歸改成迭代的方法，你需要知道迭代指針是如何走的，還要借助數據結構才能實現，就是你需要注意那些需要先放入數據結構，哪些需要后放入

下面介紹的方法是一種通用方法

以先根遍歷為例：

def preorder(root):if not root:returnelse:print(root.data,end=' ')preorder(root.left)preorder(root.right)# -*- coding: utf-8 -*-class BinTNode:def __init__(self,data=None,left =None,right =None):self.data =dataself.left =leftself.right =rightclass guide:def __init__(self,operation =None,node =None,level =None):self.operation = operationself.node =node# 針對專門的問題可以專門設計需要的恢復的現場self.level =leveldef preorder(root):# operation = 1 代表處理訪問自己，operation = 0訪問子樹stack = []result =[]# 初始化# 原始問題：訪問自己，訪問左子樹，訪問右子樹# 入棧順序：訪問右子樹，訪問左子樹，訪問自己stack.append(guide(0,root))# 這里把原問題看成一個對象，這個對象有三個部分：左子樹，當前結點，右子樹，# 這個對象可以有兩種操作，一是處理當前結點，二是訪問子樹，訪問子樹可以看成子問題while stack:Current = stack.pop()if Current.operation == 1:result.append(Current.node.data)# 為1代表訪問子樹，訪問子樹就當作子問題處理else:# 入棧順序：訪問右子樹，訪問左子樹，訪問自己if Current.node.right:stack.append(guide(0,Current.node.right))if Current.node.left:stack.append(guide(0,Current.node.left))if Current.node:stack.append(guide(1,Current.node))return resultdef inorder(root):# operation = 1 代表處理訪問自己，operation = 0訪問子樹stack = []result =[]# 初始化# 原始問題：訪問自己，訪問左子樹，訪問右子樹# 入棧順序：訪問右子樹，訪問左子樹，訪問自己stack.append(guide(0,root))# 這里把原問題看成一個對象，這個對象有三個部分：左子樹，當前結點，右子樹，# 這個對象可以有兩種操作，一是處理當前結點，二是訪問子樹，訪問子樹可以看成子問題while stack:Current = stack.pop()if Current.operation == 1:result.append(Current.node.data)# 為1代表訪問子樹，訪問子樹就當作子問題處理else:# 入棧順序：訪問右子樹，訪問自己，訪問左子樹if Current.node.right:stack.append(guide(0,Current.node.right))if Current.node:stack.append(guide(1,Current.node))if Current.node.left:stack.append(guide(0,Current.node.left))return resultdef postorder(root):# operation = 1 代表處理訪問自己，operation = 0訪問子樹stack = []result =[]# 初始化# 原始問題：訪問自己，訪問左子樹，訪問右子樹# 入棧順序：訪問右子樹，訪問左子樹，訪問自己stack.append(guide(0,root))# 這里把原問題看成一個對象，這個對象有三個部分：左子樹，當前結點，右子樹，# 這個對象可以有兩種操作，一是處理當前結點，二是訪問子樹，訪問子樹可以看成子問題while stack:Current = stack.pop()if Current.operation == 1:result.append(Current.node.data)# 為1代表訪問子樹，訪問子樹就當作子問題處理else:# 入棧順序：訪問自己,訪問右子樹，訪問左子樹，if Current.node:stack.append(guide(1,Current.node))if Current.node.right:stack.append(guide(0,Current.node.right))if Current.node.left:stack.append(guide(0,Current.node.left))return resultdef countTree(root):stack = []result = 0# 初始化樹，把樹看成待訪問的一顆子樹 stack = [guide(0,root),]while stack:pointer = stack.pop()if pointer.operation == 1:result +=1else:if pointer.node:stack.append(guide(1,pointer.node))if pointer.node.right:stack.append(guide(0,pointer.node.right))if pointer.node.left:stack.append(guide(0,pointer.node.left)) return resultdef sumTree(root):stack = []result = 0# 初始化樹，把樹看成待訪問的一顆子樹 stack = [guide(0,root),]while stack:pointer = stack.pop()if pointer.operation == 1:result +=pointer.node.dataelse:if pointer.node:stack.append(guide(1,pointer.node))if pointer.node.right:stack.append(guide(0,pointer.node.right))if pointer.node.left:stack.append(guide(0,pointer.node.left)) return result def heightTree(root):stack = []MAX = 0# 初始化樹，把樹看成待訪問的一顆子樹 stack = [guide(0,root,0)]while stack:pointer = stack.pop()if pointer.operation == 1:if not pointer.node.left and not pointer.node.left:if pointer.level > MAX:MAX = pointer.levelelse:if pointer.node:stack.append(guide(1,pointer.node,pointer.level))if pointer.node.right:stack.append(guide(0,pointer.node.right,pointer.level+1))if pointer.node.left:stack.append(guide(0,pointer.node.left,pointer.level+1)) return MAX + 1 root = BinTNode(1,BinTNode(2,BinTNode(4),BinTNode(5)),BinTNode(3,BinTNode(6),BinTNode(7)))print('preorder:',end='') print(preorder(root)) print(end='\n') print('inorder:',end='') print(inorder(root)) print(end='\n') print('postorder:',end='') print(postorder(root)) print(end='\n') print('countTree:',end='') print(countTree(root)) print(end='\n') print('sumTree:',end='') print(sumTree(root)) print(end='\n') print('heightTree:',end='') print(heightTree(root)) print(end='\n') runfile('D:/share/test/recursion_iteration_frame.py', wdir='D:/share/test') preorder:[1, 2, 4, 5, 3, 6, 7]inorder:[4, 2, 5, 1, 6, 3, 7]postorder:[4, 5, 2, 6, 7, 3, 1]countTree:7sumTree:28heightTree:3

總結

以上是生活随笔為你收集整理的树形结构：递归转化为迭代，万能通用方法，分治策略基于栈的实现的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。