from https://leetcode.com/problems/binary-tree-upside-down/;
一个列子:在翻转前
1
/ \
2 3
/ \
4 5
翻转后:
4
/ \
5 2
/ \
3 1
先从最简单的情况开始分析:如果只有一个节点,那么返回该节点就可以了(似乎太简单了);如果存在三个节点,父节点和左右子节点(如果有右节点,至少会有一个左节点),那么新的树的结构是左节点变为父节点,右节点变为新的父节点的左节点,原来的父节点变为现在的右节点;
更复杂的情况,考虑,如果左子树已经被翻转了,那么现在应该把父节点和右节点放在哪里?有点递归的味道了;从例子里可以看出,节点1和节点3需要作为节点2的子节点;而节点2是左子树翻转后的最右边的节点;所以回答放在哪里的问题,只要找到新的子树的最右的子节点,并把当前父节点和右节点连接上去就可以了;
public TreeNode upsideDownBinaryTree(TreeNode root) {
if(root == null || (root.left == null && root.right == null)) {
return root;
}
TreeNode leftRoot = upsideDownBinaryTree(root.left);
TreeNode tmpNode = leftRoot;
while(tmpNode.right != null) {
tmpNode = tmpNode.right;
}
tmpNode.left = root.right;
tmpNode.right = root;
root.right = null;
root.left = null;
return leftRoot;
}