算法通关村-----二叉树层次遍历原来如此简单

二叉树层次遍历

实现思路

利用队列先进先出的特性，首先将根元素入队，之后每次有元素出队，都要判断一下元素的左右子树是否为空，不为空，则入队，如此循环，直至队列为空。

代码实现

public List<Integer> levelTraversal(TreeNode root) {
		if(root==null){
			return new ArrayList<Integer>();
		}
		List<Integer> treeList = new ArrayList<Integer>();
        LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(root);
        while(queue.size()>0){
             treeNode = queue.remove();
             if(treeNode.right!=null){
                 queue.add(treeNode.right);
             }
             if(treeNode.left!=null){
                 queue.add(treeNode.left);
             }
             treeList.add(treeNode.val);
        }
        return treeList;
    }

处理不同层的值

实现思路

在遍历过程中，使用一个变量size记录每一层元素的个数，然后对size个数据进行处理

例题1

给定一棵二叉树的根节点，寻找每层的最大值，详见leetcode515

代码实现

public List<Integer> largestValues(TreeNode root) {
        if(root==null){
            return new ArrayList<Integer>();
        }
        List<Integer> list = new ArrayList<Integer>();
        LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(root);
        while(queue.size()>0){
            int size = queue.size();
            int max = Integer.MIN_VALUE;
            while(size>0){
                TreeNode treeNode = queue.remove();
                size--;
                int value = treeNode.val;
                max = value>max?value:max;
                if(treeNode.left!=null){
                    queue.add(treeNode.left);
                }
                if(treeNode.right!=null){
                    queue.add(treeNode.right);
                }
            }
            list.add(max);
        }
        return list;
    }

例题2

给定一棵二叉树的根节点，寻找每层的平均值,详见leetcode637

代码实现

public List<Double> averageOfLevels(TreeNode root) {
        List<Double> list = new ArrayList<Double>();
        LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(root);
        while(queue.size()>0){
            int size = queue.size();
            int sum = 0;
            int count = size;
            while(size>0){
                TreeNode treeNode = queue.remove();
                size--;
                sum += treeNode.val;
                if(treeNode.left!=null){
                    queue.add(treeNode.left);
                }
                if(treeNode.right!=null){
                    queue.add(treeNode.right);
                }
            }
            double num = sum/(double)count;
            list.add(num);
        }
        return list;
    }

例题3

给定一个二叉树的 根节点 root，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。详见leetcode199

代码实现

public List<Integer> rightSideView(TreeNode root) {
        if(root==null){
            return new ArrayList<Integer>();
        }
        List<Integer> list = new ArrayList<Integer>();
        LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(root);
        while(queue.size()>0){
            int size = queue.size();
            while(size>0){
                TreeNode treeNode = queue.remove();
                size--;
                if(size==0){
                    list.add(treeNode.val);
                }
                if(treeNode.left!=null){
                    queue.add(treeNode.left);
                }
                if(treeNode.right!=null){
                    queue.add(treeNode.right);
                }
            }
        }
        return list;
    }

例题4

给定一个二叉树的 根节点 root，请找出该二叉树的 最底层 最左边 节点的值。假设二叉树中至少有一个节点。详见leetcode513

代码实现

public int findBottomLeftValue(TreeNode root) {
        LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
        queue.add(root);
        TreeNode treeNode = new TreeNode(0);
        while(queue.size()>0){
            int size = queue.size();
            while(size>0){
                treeNode = queue.remove();
                size--;
                if(treeNode.right!=null){
                    queue.add(treeNode.right);
                }
                if(treeNode.left!=null){
                    queue.add(treeNode.left);
                }
            }
        }
        return treeNode.val;
    }