举例让抽象具体化 从上往下打印二叉树

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;

/**
 * 题目:从上往下打印出二叉树的每个节点,同层节点从左至右打印。
 *
 * 思路:
 * 1、寻找规律,发现可以使用队列解决问题
 * 2、先找到头结点
 * 3、使用循环弹出队头,然后压入左节点右节点
 * 4、注意判空
 */

public class Solution {
    public class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;

        }
    }

    public static void main(String[] args) {


    }

    public static ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        if(root == null){
            return null;
        }

        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        ArrayList<Integer> list = new ArrayList<>();
        //找不到次次数,用while
        while (!queue.isEmpty()){
            //找到当前节点 (key)
            TreeNode treeNode = queue.poll();
            if (treeNode.left != null){
                queue.offer(treeNode.left);
            }
            if (treeNode.right != null){
                queue.offer(treeNode.right);
            }
            list.add(treeNode.val);
        }
        return list;
    }
}

猜你喜欢

转载自blog.csdn.net/qq18938308968/article/details/85317867