leetcode971. 翻转二叉树以匹配先序遍历(dfs)

115 阅读1分钟

给定一个有 N 个节点的二叉树,每个节点都有一个不同于其他节点且处于 {1, …, N} 中的值。

通过交换节点的左子节点和右子节点,可以翻转该二叉树中的节点。

考虑从根节点开始的先序遍历报告的 N 值序列。将这一 N 值序列称为树的行程。

(回想一下,节点的先序遍历意味着我们报告当前节点的值,然后先序遍历左子节点,再先序遍历右子节点。)

我们的目标是翻转最少的树中节点,以便树的行程与给定的行程 voyage 相匹配。

如果可以,则返回翻转的所有节点的值的列表。你可以按任何顺序返回答案。

如果不能,则返回列表 [-1]。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int pos=0;
    List<Integer> r=new ArrayList<>();
    public List<Integer> flipMatchVoyage(TreeNode root, int[] voyage) {
              flip(root,voyage);
         return r;
    }
    public void flip(TreeNode root,int[] voyage) {
             if(root==null||r.size()>0&&r.get(0)==-1) return;//如果出现了不满足的情况或者节点为空直接返回
        if(root.val!=voyage[pos])//交换完还是不符合则返回-1
        {
            r.clear();
            r.add(-1);
            return ;
        }
         if(root.left!=null&&root.left.val!=voyage[pos+1])//如果不符合则交换左右子树
         {
             TreeNode temp=root.left;
             root.left=root.right;
             root.right=temp;
             r.add(root.val);//将需要交换子树的节点入队
         }
         pos++;
         flip(root.left,voyage);
         flip(root.right,voyage);
    }
}