版权声明:本文为博主原创文章,未经博主允许不得转载。有事联系:[email protected] https://blog.csdn.net/qq_17550379/article/details/85987595
给定一个有 N
个节点的二叉树,每个节点都有一个不同于其他节点且处于 {1, ..., N}
中的值。
通过交换节点的左子节点和右子节点,可以翻转该二叉树中的节点。
考虑从根节点开始的先序遍历报告的 N
值序列。将这一 N
值序列称为树的行程。
(回想一下,节点的先序遍历意味着我们报告当前节点的值,然后先序遍历左子节点,再先序遍历右子节点。)
我们的目标是翻转最少的树中节点,以便树的行程与给定的行程 voyage
相匹配。
如果可以,则返回翻转的所有节点的值的列表。你可以按任何顺序返回答案。
如果不能,则返回列表 [-1]
。
示例 1:
输入:root = [1,2], voyage = [2,1]
输出:[-1]
示例 2:
输入:root = [1,2,3], voyage = [1,3,2]
输出:[1]
示例 3:
输入:root = [1,2,3], voyage = [1,2,3]
输出:[]
提示:
1 <= N <= 100
解题思路
这个问题非常简单,我们只要遍历root
,然后遍历voyage
,比较两者的val
是不是一样,不一样的话,自然就不对,如果一样的话,继续遍历下去即可。例如
我们发现此时的root.val
和voyage[0]
一致,所以我们比较root.left
和root.right
。
我们发现此时root.left
和voyage[1]
不一样,所以我们需要将root.left
和root.right
交换。
然后继续判断root.left
和voyage[1]
,并且判断voyage[2]
和root.right
。
class Solution:
def flipMatchVoyage(self, root, voyage):
"""
:type root: TreeNode
:type voyage: List[int]
:rtype: List[int]
"""
res = list()
i = 0
def dfs(root):
nonlocal i
if not root:
return True
if root.val != voyage[i]:
return False
i += 1
if root.left and root.left.val != voyage[i]:
res.append(root.val)
root.left, root.right = root.right, root.left
return dfs(root.left) and dfs(root.right)
return res if dfs(root) else [-1]
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!