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

主要代码

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。