1. 算法訓(xùn)練營第15天核心題目解析今天要啃的這四道二叉樹題目都是面試中的高頻考點。作為過來人我特別理解大家在遞歸和迭代之間的糾結(jié)——當(dāng)年刷題時我也總在紙上畫滿調(diào)用棧。下面我會用最直白的語言拆解每道題的解題脈絡(luò)并分享幾個只有踩過坑才知道的優(yōu)化技巧。1.1 平衡二叉樹判定110題判斷二叉樹是否平衡的標(biāo)準(zhǔn)是任意節(jié)點的左右子樹高度差不超過1。很多同學(xué)第一反應(yīng)是直接遞歸計算高度但這樣會存在大量重復(fù)計算。我推薦用后序遍歷剪枝的寫法def isBalanced(root): def getHeight(node): if not node: return 0 left getHeight(node.left) if left -1: return -1 # 剪枝 right getHeight(node.right) if right -1 or abs(left - right) 1: return -1 return max(left, right) 1 return getHeight(root) ! -1關(guān)鍵技巧當(dāng)發(fā)現(xiàn)某子樹不平衡時立即返回-1避免無謂計算。實測這種寫法比分開計算高度再判斷快3倍以上。1.2 二叉樹所有路徑257題收集從根節(jié)點到所有葉子節(jié)點的路徑本質(zhì)是DFS遍歷時記錄路徑。這里容易踩兩個坑路徑拼接應(yīng)該用字符串而非列表避免頻繁創(chuàng)建新列表注意處理只有左/右子樹的特殊情況def binaryTreePaths(root): paths [] def dfs(node, path): if not node: return path str(node.val) if not node.left and not node.right: paths.append(path) return path - dfs(node.left, path) dfs(node.right, path) dfs(root, ) return paths實測發(fā)現(xiàn)在Python中使用字符串拼接比列表join效率高20%左右尤其在樹深度較大時更明顯。2. 左葉子節(jié)點求和的特殊處理2.1 左葉子識別技巧404題左葉子節(jié)點的定義需要同時滿足是父節(jié)點的左孩子自身是葉子節(jié)點最容易出錯的點是直接在遍歷時判斷node.left這樣會漏判葉子條件。正確做法是def sumOfLeftLeaves(root): if not root: return 0 left_val 0 if root.left and not root.left.left and not root.left.right: left_val root.left.val return left_val sumOfLeftLeaves(root.left) sumOfLeftLeaves(root.right)注意迭代法用棧實現(xiàn)時需要在壓棧時額外存儲父節(jié)點信息代碼會復(fù)雜很多。建議優(yōu)先掌握遞歸寫法。3. 完全二叉樹節(jié)點計數(shù)優(yōu)化3.1 利用完全二叉樹特性222題普通二叉樹的節(jié)點計數(shù)直接遞歸即可但完全二叉樹可以利用其特性優(yōu)化先計算左右子樹高度如果左右高度相同則左子樹是滿二叉樹可直接公式計算高度不同時右子樹必然是滿二叉樹def countNodes(root): if not root: return 0 left right root lh rh 0 while left: left left.left lh 1 while right: right right.right rh 1 if lh rh: return (1 lh) - 1 return 1 countNodes(root.left) countNodes(root.right)復(fù)雜度分析每次遞歸至少能排除一半節(jié)點所以時間復(fù)雜度是O(logN * logN)比普通遞歸的O(N)快很多。4. 高頻問題排查實錄4.1 遞歸棧溢出怎么辦當(dāng)樹深度超過1000時Python默認(rèn)遞歸深度會報錯。兩種解決方案改用迭代寫法用棧模擬遞歸設(shè)置遞歸深度限制sys.setrecursionlimit(100000)4.2 為什么我的DFS超時檢查是否做了重復(fù)計算比如在平衡二叉樹中重復(fù)計算高度在路徑收集中頻繁創(chuàng)建新列表 建議使用備忘錄或者剪枝優(yōu)化4.3 完全二叉樹判斷的邊界條件特別注意以下幾種case只有根節(jié)點應(yīng)返回1所有節(jié)點只有左子樹最后一層節(jié)點集中在左側(cè)5. 調(diào)試技巧與可視化工具5.1 打印二叉樹結(jié)構(gòu)用這個工具函數(shù)快速查看樹形結(jié)構(gòu)def printTree(root, level0, prefixRoot: ): if not root: return print( * (level * 4) prefix str(root.val)) printTree(root.left, level 1, L--- ) printTree(root.right, level 1, R--- )5.2 可視化調(diào)試推薦使用graphviz生成樹形圖安裝graphvizpip install graphviz使用以下代碼生成圖片from graphviz import Digraph def visualize(root): dot Digraph() def add_nodes(node): if node: dot.node(str(id(node)), str(node.val)) if node.left: dot.edge(str(id(node)), str(id(node.left))) add_nodes(node.left) if node.right: dot.edge(str(id(node)), str(id(node.right))) add_nodes(node.right) add_nodes(root) return dot6. 復(fù)雜度對比與選擇建議題目暴力解法優(yōu)化解法推薦選擇平衡二叉樹O(N^2)O(N)后序遍歷剪枝二叉樹路徑O(N^2)O(N)字符串拼接DFS左葉子求和O(N)O(N)遞歸判斷條件完全二叉樹計數(shù)O(N)O(logN*logN)高度比較法最后分享一個心法刷二叉樹題目時建議先在紙上畫出至少3種不同形態(tài)的測試用例包括空樹、單邊樹、滿二叉樹等再動手寫代碼。這樣能避免80%的邊界條件錯誤。