[每日一道小算法(三十)] [树(队列或栈)] 按之字型打印二叉树

前言:
哇,今天情人节啊,我这条单身狗只能敲代码来缓解一下啦,哈哈。

题目描述

请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。

解题思路

实现这道题的方法其实有很多,我就说一下我自己的这道题思路吧。我是使用队列来完成,有很多人都是使用栈来完成,思路其实都一样。
我们使用一个队列来保存每一层的节点,根据奇数偶数层来判断正序打印还是逆序打印。就是这样一个思路,直接看代码吧,这样你们理解起来更好一些。

代码样例

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/
public class Solution {
public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
        if(pRoot == null)
        {
            return result;
        }
        ArrayList<Integer> list = null;
        //队列来保存每层节点
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        int rows = 1;
        queue.add(pRoot);
        while(!queue.isEmpty())
        {
            list = new ArrayList<Integer>();
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                if(rows%2==0)
                {
                    list.add(0,node.val);
                }else {
                    list.add(node.val);
                }
                if(node.left!=null)
                {
                    queue.offer(node.left);
                }
                if(node.right!=null)
                {
                    queue.offer(node.right);
                }
            }
            result.add(list);
            rows++;
        }
        return result;
    }

}
发布了157 篇原创文章 · 获赞 34 · 访问量 4383

猜你喜欢

转载自blog.csdn.net/qq_39397165/article/details/104312507
今日推荐