题目地址(971. 翻转二叉树以匹配先序遍历)

题目地址(971. 翻转二叉树以匹配先序遍历)

https://leetcode.cn/problems/flip-binary-tree-to-match-preorder-traversal/

题目描述

给你一棵二叉树的根节点 root ，树中有 n 个节点，每个节点都有一个不同于其他节点且处于 1 到 n 之间的值。

另给你一个由 n 个值组成的行程序列 voyage ，表示 预期 的二叉树 先序遍历 结果。

通过交换节点的左右子树，可以 翻转 该二叉树中的任意节点。例，翻转节点 1 的效果如下：

请翻转 最少 的树中节点，使二叉树的 先序遍历 与预期的遍历行程 voyage 相匹配 。 

如果可以，则返回 翻转的 所有节点的值的列表。你可以按任何顺序返回答案。如果不能，则返回列表 [-1]。

 

示例 1：

输入：root = [1,2], voyage = [2,1]
输出：[-1]
解释：翻转节点无法令先序遍历匹配预期行程。


示例 2：

输入：root = [1,2,3], voyage = [1,3,2]
输出：[1]
解释：交换节点 2 和 3 来翻转节点 1 ，先序遍历可以匹配预期行程。

示例 3：

输入：root = [1,2,3], voyage = [1,2,3]
输出：[]
解释：先序遍历已经匹配预期行程，所以不需要翻转节点。


 

提示：

树中的节点数目为 n
n == voyage.length
1 <= n <= 100
1 <= Node.val, voyage[i] <= n
树中的所有值 互不相同
voyage 中的所有值 互不相同

前置知识

公司

• 暂无

思路

关键点

代码

• 语言支持：Python3

Python3 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 flipMatchVoyage(self, root: Optional[TreeNode], voyage: List[int]) -> List[int]:
        """
        翻转结点，让树的先序遍历符合voyage顺序
        """
        self.fliped = []
        self.i = 0

        def dfs(root):
            if root:
                if root.val != voyage[self.i]:
                    self.fliped = [-1]
                    return 
                self.i += 1
                # 决定递归的顺序
                if (self.i < len(voyage) and 
                root.left and root.left.val != voyage[self.i]):
                    self.fliped.append(root.val)
                    dfs(root.right)
                    dfs(root.left)
                else:
                    dfs(root.left)
                    dfs(root.right)
        dfs(root)
        if -1 in self.fliped:
            return [-1]
        else:
            return self.fliped

复杂度分析

令 n 为数组长度。

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$