原题链接: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来解,这题比较特殊