原题链接:https://leetcode.com/problems/binary-tree-upside-down/
这是一道带锁题,原题是说给定一个二叉树,二叉树的右子树要么是空的,要么是有一个兄弟节点的叶子子树(没有孩子)。题目要求讲这课二叉树上下翻转一下,并且使得二叉树的右节点变成左节点。
解:
这个树是一棵有特征的树,它的生长是沿着左子树方向的。所以我们只需定义好几个变量,沿着左子树将每个节点迭代处理即可:假设node,将node的右子树设为原树的parent节点,讲node的左子树设为原树中的兄弟右节点即可。时间复杂度为O(h),h为树的高度。代码如下:
public TreeNode upsideDownBinaryTree(TreeNode root) { if(root == null) return null; TreeNode parent = null; TreeNode node = root; TreeNode right = null; while(node!=null) { TreeNode left = node.left; node.left = right; right = node.right; node.right = parent; parent = node; node = left; } return parent; }
遇到树的问题,主要还是用DFS来解,这题比较特殊