上次和波波聊天,说到面试的编程题的问题,他提到google有一个经典的面试题:将二叉树左右子树颠倒。
然后我有一次面试就让一个候选人写一下,结果他竟然直接放弃了!
这题目其实挺简单的:
-
首先写一个类,用来封装我们的代码:
/**
* @author jingxuan
* @date 2021/1/11 8:09 下午
*/
public class MirrorReverse {
public static void main(String[] args) {
}
}
-
简单定义个二叉树的结构,方便起见可以直接用public修饰:
public class MirrorReverse {
static class Node {
public int value;
public Node left;
public Node right;
}
}
-
实现一下逻辑。算法其实非常简单,只需要满足两个条件就可以了:
因此,使用深度优先遍历、广度优先遍历都没有关系,哪怕是啥奇怪的随机遍历算法都没事,只要保证上面两点要求都满足就好了。
-
所有节点会且仅会被遍历一次;
-
所有节点的左右子树会且仅会被交换一次;
-
public static Node reverse(Node root) {
if (root == null) return root;
// 交换
Node tmp = root.left;
root.left = root.right;
root.right = tmp;
// 递归
reverse(root.left);
reverse(root.right);
return root;
}
-
有时候可能会要求写非递归的算法,其实也非常简单:
public static Node reverse(Node root) {
if (root == null) return root;
// 方便垃圾回收器尽快收集垃圾,优先使用FIFO类型的队列
Queue<Node> nodes = new LinkedList<>();
nodes.offer(root);
while (nodes.size() != 0) {
Node node = nodes.remove();
Node tmp = node.left;
node.left = node.right;
node.right = tmp;
if (node.left != null) nodes.offer(node.left);
if (node.right != null) nodes.offer(node.right);
}
return root;
}
嗯,其实真的挺简单的……