二叉树中,如果一个节点是其父节点的唯一子节点,则称这样的节点为 “独生节点” 。二叉树的根节点不会是独生节点,因为它没有父节点。
给定一棵二叉树的根节点 root ,返回树中 所有的独生节点的值所构成的数组 。数组的顺序 不限 。
示例 1:
输入:root = [1,2,3,null,4]
输出:[4]
解释:浅蓝色的节点是唯一的独生节点。
节点 1 是根节点,不是独生的。
节点 2 和 3 有共同的父节点,所以它们都不是独生的。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-all-the-lonely-nodes
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
——————————
解题思路:使用深度优先遍历,对于二叉树中的某一个结点,查询其左子树和右子树,如果左子树或者右子树为空,则记录左子树和右子树中不为空的值。
# 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 __init__(self):
self.ans = [] # 用于保存最后的结果
def getLonelyNodes(self, root: TreeNode) -> List[int]:
def find_node(root):
if not root: # 如果当前结点为空,返回None
return None
left = find_node(root.left) # 当前结点左子结点的值
right = find_node(root.right) # 当前结点右子结点的值
if not left and right: # 如果左子结点为空,右子结点不为空,记录右子结点的值
self.ans.append(right)
elif not right and left: # 如果右子结点为空,左子结点不为空,记录左子结点的值
self.ans.append(left)
return root.val # 返回当前结点的值
find_node(root)
return self.ans
需要遍历每个结点,所以时间复杂度为O(n),使用深度变量,需要保存中间变量,所以空间复杂度为O(logn)。