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; }
时间: 2024-10-20 09:00:56