思路:
开始没写出来
思路:用层序遍历,每一层交换结点
问题描述
用层序遍历,每一层交换结点,但是交换后的结点会影响下一层的结点,结果出错
错误代码
public TreeNode invertTree(TreeNode root) {
//翻转链表,基本上就是交换数字的位置,用层序遍历可以,但是广度搜索效率比较低
if(root == null){
return root;
}
Deque<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while(!queue.isEmpty()){
int n = queue.size();
Deque<TreeNode> queue1 = new LinkedList<>();
for(int i =0; i< n; i++){
TreeNode temp = queue.poll();
if(temp.left != null){
queue1.offer(temp.left);
}
if(temp.right != null){
queue1.offer(temp.right);
}
queue.offer(temp);
}
//交换 智能用另一个链表,要不然弹出后就找不到了
for(int j = 0; j < n; j++){
TreeNode temp = queue.poll();
if(temp.left != null){
temp.left.val = queue1.pollLast().val;
queue1.offerFirst(temp.left);
}
if(temp.right != null){
temp.right.val = queue1.pollLast().val;
queue1.offerFirst(temp.right);
}
}
queue = queue1;
}
return root;
}
正确代码
class Solution {
public TreeNode invertTree(TreeNode root) {
if(root==null) {
return null;
}
//将二叉树中的节点逐层放入队列中,再迭代处理队列中的元素
LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
queue.add(root);
while(!queue.isEmpty()) {
//每次都从队列中拿一个节点,并交换这个节点的左右子树
TreeNode tmp = queue.poll();
TreeNode left = tmp.left;
tmp.left = tmp.right;
tmp.right = left;
//如果当前节点的左子树不为空,则放入队列等待后续处理
if(tmp.left!=null) {
queue.add(tmp.left);
}
//如果当前节点的右子树不为空,则放入队列等待后续处理
if(tmp.right!=null) {
queue.add(tmp.right);
}
}
//返回处理完的根节点
return root;
}
}
原因分析:
本来只要单层之间交换结点就行了,思路错了,上下层互相影响了
解决方案:
只要交换每个结点左右子树就可以实现翻转的效果,这谁能想到啊