数据结构专题——二叉树的层次遍历（三种形式）

###二叉树专题——二叉树的层次遍历

定义

class TreeNode{
	int data;
	TreeNode left;
	TreeNode right;	
}

二叉树的层次遍历多种形式

第一种：层次遍历不需要分层（最简单）

public static List<Integer> simpleLevelTraversal(TreeNode root) {
		
		List<Integer> list = new ArrayList<Integer>();		
		if (root == null) {
			return new ArrayList<Integer>();
		}
    	// 反复入队出队的逻辑 可以画图理解
		Queue<TreeNode> queue = new LinkedList<TreeNode>();
		queue.offer(root);
		while (!queue.isEmpty()) {
			
			TreeNode node = queue.poll();
			list.add(node.data);
			if (node.left != null) {
				queue.offer(node.left);
			}
			if (node.right != null) {
				queue.offer(node.right);
			}
		}
		return list;
	}

第二种：从上到下层次遍历，每层单独存储

public static List<List<Integer>> LevelTraversal(TreeNode root) {
		
		if (root == null) {
			return new ArrayList<List<Integer>>();
		}
		List<List<Integer>> res = new ArrayList<List<Integer>>();
		Queue<TreeNode> queue = new LinkedList<TreeNode>();
		queue.add(root);		
		while (!queue.isEmpty()) {
            // 核心：提前保存该层的个数
			int size = queue.size();
			List<Integer> temp = new ArrayList<>();			
			for (int i = 0; i < size; i++) {
				TreeNode node = queue.poll();
				temp.add(node.data);
				if (node.left != null) {
					queue.add(node.left);
				}
				if (node.right != null) {
					queue.add(node.right);
				}
			}
			res.add(temp);
		}
		return res;	
	}

第三种：自底向上分层遍历

public static List<List<Integer>> LevelTraversalFromBottom(TreeNode root) {
		
		if (root == null) {
			return new LinkedList<List<Integer>>();
		}
		List<List<Integer>> res = new LinkedList<List<Integer>>();		
		Deque<TreeNode> queue = new LinkedList<TreeNode>();
		queue.add(root);
		while (!queue.isEmpty()) {
			
			int size = queue.size();	
            // 提前准备好list，因为我们不能正常顺序插入了
			List<Integer> temp = new LinkedList<Integer>();		
			for (int i = 0; i < size; i++) {
				TreeNode node =  queue.poll();
				temp.add(node.data);
				if (node.left != null) {
					queue.add(node.left);
				}
				if (node.right != null) {
					queue.add(node.right);
				}
			}
            // 核心部分：这次通过temp尾插法，达到我们要的效果
			res.add(0,temp);
		}
		return res;
		
	}