※ 引述《argorok (死肥肥社管)》之銘言： : : https://leetcode.com/problems/add-one-row-to-tree : : 623. Add One Row to Tree : : 給你一個二元樹，請在深度為depth的位置插入一列值為val的節點。 : 今天遞迴解起來還蠻順的 : 感覺有點手感了 : class Solution(object): : def addOneRow(self, root, val, depth): : """ : :type root: TreeNode : :type val: int : :type depth: int : :rtype: TreeNode : """ : if depth == 1: : return TreeNode(val, left=root) : if depth == 2: : if root: : tmpLeft = TreeNode(val, left=root.left) : tmpRight = TreeNode(val, right=root.right) : root.left = tmpLeft : root.right = tmpRight : elif depth > 2: : if root.left: : self.addOneRow(root.left, val, depth-1) : if root.right: : self.addOneRow(root.right, val, depth-1) : return root 思路: bfs找到指定層數 然後把東西放進去 假設指定層數為一 直接在root前面放東西就好 Python Code: # Definition for a binary tree node. # class TreeNode: # def __init__(self, val=0, left=None, right=None): # self.val = val # self.left = left # self.right = right class Solution: def addOneRow(self, root: Optional[TreeNode], val: int, depth: int) -> Optional[TreeNode]: if depth == 1: root = TreeNode(val,root,None) return root q = deque() q.append(root) h = 1 while q: l = len(q) for i in range(l): node = q.popleft() if h == depth - 1: node.left = TreeNode(val,node.left,None) node.right = TreeNode(val,None,node.right) if node.left: q.append(node.left) if node.right: q.append(node.right) h += 1 return root -- ※ 發信站: 批踢踢實業坊(ptt.org.tw), 來自: 114.43.137.148 (臺灣) ※ 文章網址: https://ptt.org.tw/Marginalman/M.1713240445.A.E64

推 digua: 大師 04/16 12:10

推 SecondRun: 大師 04/16 12:13

推 JIWP: 別卷了 04/16 12:21

→ DJYOSHITAKA: 剩我寫不了了 04/16 12:23