java - 改进广度优先逻辑

Question

不久前我接到了一个任务。我的教授评分后，他在代码中留下了一条评论:“简化逻辑”。他没有取消任何观点，但我看不到其他方法来解决这个问题……我所拥有的对我来说是有意义的，并且有效。那么，有人可以告诉我一种方法来改善我所拥有的吗？据我所知...

    public void breadthFirstTraversal(){
        breadthFirstTraversal(root);
    }

    private void breadthFirstTraversal(TreeNode<E> node){

        Queue<TreeNode<E>> queue = new LinkedList<>();

        queue.add(node);

        while(!queue.isEmpty()){
            TreeNode<E> temp = queue.poll();
            System.out.print(temp.data + " ");

            if(temp.left != null && temp.right == null){
                queue.add(temp.left);
            }else if(temp.left == null && temp.right != null){
                queue.add(temp.right);
            }else if(temp.left != null && temp.right != null){
                queue.add(temp.left);
                queue.add(temp.right);
            }
        }
    }

提前谢谢您!

Answer 1

temp.right 为 null 对于是否应添加 temp.left 没有影响，而 temp.left 为 null 则没有影响对添加 temp.right 的影响。您可以将其减少到只有两个 if:

if( temp.left != null ) queue.add( temp.left );
if( temp.right != null ) queue.add( temp.right );