971. 翻转二叉树以匹配先序遍历

971. 翻转二叉树以匹配先序遍历

原始题目链接：https://leetcode-cn.com/problems/flip-binary-tree-to-match-preorder-traversal/
给你一棵二叉树的根节点 root ，树中有 n 个节点，每个节点都有一个不同于其他节点且处于 1 到 n 之间的值。另给你一个由 n 个值组成的行程序列 voyage ，表示 预期 的二叉树 先序遍历 结果。通过交换节点的左右子树，可以 翻转 该二叉树中的任意节点。例，翻转节点 1 的效果如下：

解题思路：

题目要求先序遍历二叉树的时候，遇到和行程序列不一样的值，通过翻转来实现一样，如果翻转也不能实现一样，则返回[-1]，所以在翻转二叉树的时候需要记录翻转的节点到一个列表中，最后返回这个列表。翻转二叉树在之前的题中有做过，可以用两种方式实现，一种是递归，一种是迭代，核心都是交换左右子树。本题使用递归，设计一个递归函数，用于翻转二叉树并且记录翻转的节点。具体实现看代码及注释。

代码实现：

# 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 = []
        # 行程序列索引
        self.index = 0

    # 深度优先遍历函数
    def dfs(self, root, voyage):
        if root:
            # 递归结束条件：如果遍历到的节点（翻转或未翻转遍历到的节点）不能与行程序列的值相匹配，则匹配失败，返回-1
            if root.val != voyage[self.index]:
                self.ans = [-1]
                return
            
            # 开始深度优先遍历
            self.index += 1
            # 因为题目要求是先序遍历匹配，如果当前节点的左子树的值和行程序列的值不相等，就翻转二叉树
            # 翻转的时候记录节点值，保存在ans数组中
            if (self.index < len(voyage)) and root.left and root.left.val != voyage[self.index]:
                self.ans.append(root.val)
                # 翻转二叉树
                self.dfs(root.right, voyage)
                self.dfs(root.left, voyage)
            # 正常先序遍历二叉树的情况
            else:
                self.dfs(root.left, voyage)
                self.dfs(root.right, voyage)

    def flipMatchVoyage(self, root: TreeNode, voyage: List[int]) -> List[int]:
        self.dfs(root, voyage)
        if self.ans and self.ans[0] == -1:
            self.ans = [-1]
        return self.ans

参考文献：
https://leetcode-cn.com/problems/flip-binary-tree-to-match-preorder-traversal/solution/fan-zhuan-er-cha-shu-yi-pi-pei-xian-xu-bian-li-by-/