226. 翻转二叉树
https://leetcode-cn.com/problems/invert-binary-tree/
这题真的是很经典,我们先来看看具体的做法,就是根节点(4)的两个叶子节点两两交换,然后7这个节点的两个叶子节点也是两两交换,着我们就大概得到一个过程:就是每当走过一个节点的时候,两两交换他们的叶子节点,然后再去遍历他们的叶子节点进行相同的操作步骤,两两交换他们的叶子节点,这样当然你遍历完了整个树,就会得到一个翻转之后的树。
这里引用《剑指Offer 第二版》中的一个过程图来说明,具体的翻转的过程见下图:
(a):交换根节点两个叶子节点
(b):交换10的节点的两个叶子节点
(c):交换6的节点的两个叶子节点
了解了步骤,我们再来看看具体的遍历方式,因为每次都是先交换根节点的两个叶子节点,那么我们就要先遍历根节点,那么就是要用先序遍历了。
class Solution {
public TreeNode invertTree(TreeNode root) {
if (root == null) {
return root;
}
invertOpt(root);
return root;
}
public void invertOpt (TreeNode node) {
if (node.right != null || node.left != null) {
TreeNode temp = node.left;
node.left = node.right;
node.right = temp;
} else {
return;
}
if (node.left != null) {
invertOpt(node.left);
}
if (node.right != null) {
invertOpt(node.right);
}
}
}