二叉树层次遍历并打印行号——java

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/DR_eamMer/article/details/81879335

主要代码

public void levelOrderNew(TreeNode root) {
        TreeNode last=root; //当前行最右结点
        TreeNode nlast = null;  //下一行最右结点
        TreeNode temp;
        Queue<TreeNode> queue=new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            temp=queue.poll();
            System.out.print(temp.getData()+" ");
            if(temp.getLeftNode()!=null) {
                queue.offer(temp.getLeftNode());
                nlast=temp.getLeftNode();
            }
            if(temp.getRightNode()!=null)
            {
                queue.offer(temp.getRightNode());
                nlast=temp.getRightNode();
            }
            if(temp==last) {
                System.out.println();
                last=nlast;
            }

        }
    }

算法思想:
1.分别标识出所在行的最右结点last,以及下一行的最右结点nlast;
2.nlast一直跟踪记录队列中最新加入的结点,最新加入的结点一定是发现的下一行的最右结点。所以当前行打完时,nlast一定是下一行的最右结点。
2.如果输出结点是last结点,说明该换行了,last=nlast。

猜你喜欢

转载自blog.csdn.net/DR_eamMer/article/details/81879335