劍指Offer:面試題23——從上往下打印二叉樹(java實現)

問題描述:

從上往下打印出二叉樹的每個節點,同層節點從左至右打印。

思路:

按照層次遍歷的方法,使用隊列輔助。
1.將根結點加入隊列。
2.循環出隊,打印當前元素,若該結點有左子樹,則將其加入隊列,若有右子樹,將其加入隊列。
3.直到隊列爲空,表明已經打印完所有結點。

代碼:

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

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

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

    }

}

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> result = new ArrayList<Integer>();
        if(root == null){
            return result;
        }

        Queue<TreeNode> q = new LinkedList<TreeNode>();

        q.add(root);        
        while(!q.isEmpty()){
            TreeNode tmp = q.remove();
            result.add(tmp.val);
            if(tmp.left != null){
                q.add(tmp.left);
            }

            if(tmp.right != null){
                q.add(tmp.right);

            }

        }

        return result;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章