LCP 44. 開(kāi)幕式焰火

Ideas

樹(shù)類型的題目一般都需要用到遞歸，這道題相對(duì)來(lái)說(shuō)比較簡(jiǎn)單，我們只需要遍歷整棵樹(shù)，然后把節(jié)點(diǎn)值記錄到一個(gè)集合中，最后返回集合的長(zhǎng)度就可以了。

所以這道題的考點(diǎn)就是樹(shù)的遍歷，奉上我的二叉樹(shù)前序遍歷模板：

遞歸版本：

def preorderTraversalRecursion(node):if not node:returnprint(node.value, end=' ') # 遞歸序第一次到達(dá) node 的時(shí)候 printBinaryTreeNode.preorderTraversalRecursion(node.left)BinaryTreeNode.preorderTraversalRecursion(node.right)

非遞歸版本：

def preorderTraversalLoop(node):if not node:returnstack = [node] # list 模擬 stackwhile stack:tmp = stack.pop()print(tmp.value, end=' ')if tmp.right:stack.append(tmp.right)if tmp.left:stack.append(tmp.left)

這道題把模板稍微一改就行了。

Code

Python

# Definition for a binary tree node. class TreeNode:def __init__(self, x):self.val = xself.left = Noneself.right = Noneclass Solution:def numColor(self, root: TreeNode) -> int:def pre_order_traversal(node: TreeNode):if not node:returnstack = [node]while stack:item = stack.pop()ans.add(item.val)if item.right:stack.append(item.right)if item.left:stack.append(item.left)ans = set()pre_order_traversal(root)return len(ans)

總結(jié)

以上是生活随笔為你收集整理的LeetCode Algorithm LCP 44. 开幕式焰火的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。