leetcode翻转二叉树层序遍历出错

思路:

开始没写出来

思路:用层序遍历,每一层交换结点


问题描述

用层序遍历,每一层交换结点,但是交换后的结点会影响下一层的结点,结果出错

错误代码

 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;
	}
}



原因分析:

本来只要单层之间交换结点就行了,思路错了,上下层互相影响了


解决方案:

只要交换每个结点左右子树就可以实现翻转的效果,这谁能想到啊

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值