本文共 3727 字,大约阅读时间需要 12 分钟。
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
和以往简单的BFS不同,这里要求返回结果的格式,需要处理一下
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */class Solution { public List
> levelOrder(TreeNode root) { List
> res = new ArrayList<>(); if(root == null){ return res; } Queue mq = new LinkedList<>(); mq.offer(root); while(!mq.isEmpty()){ List ml = new ArrayList<>(); int len = mq.size(); for(int i = 1;i<=len;i++){ TreeNode node = mq.poll(); ml.add(node.val); if(node.left != null){ mq.offer(node.left); } if(node.right != null){ mq.offer(node.right); } } res.add(ml); } return res; }}
对于返回结果的要求,只需要处理将 子list 每次插入到 父list 的 0 位置 即可
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */class Solution { public List
> levelOrderBottom(TreeNode root) { List
> res = new ArrayList<>(); Queue mq = new LinkedList<>(); if(root == null){ return res; } mq.offer(root); while(!mq.isEmpty()){ List ml = new ArrayList<>(); int len = mq.size(); for(int i = 1;i<=len;i++){ TreeNode node = mq.poll(); ml.add(node.val); if(node.left != null){ mq.offer(node.left); } if(node.right != null){ mq.offer(node.right); } } //这里处理一下要求返回的格式,每次将子List插到 0 位置 res.add(0,ml); } return res; }}
再贴一下C++的代码
class Solution { public: vector> levelOrderBottom(TreeNode* root) { auto levelOrder = vector >(); if (!root) { return levelOrder; } queue q; q.push(root); while (!q.empty()) { auto level = vector (); int size = q.size(); for (int i = 0; i < size; ++i) { auto node = q.front(); q.pop(); level.push_back(node->val); if (node->left) { q.push(node->left); } if (node->right) { q.push(node->right); } } levelOrder.push_back(level); } reverse(levelOrder.begin(), levelOrder.end()); return levelOrder; }};
void bfs(TreeNode root) { Queuequeue = new ArrayDeque<>(); queue.add(root); while (!queue.isEmpty()) { TreeNode node = queue.poll(); // Java 的 pop 写作 poll() if (node.left != null) { queue.add(node.left); } if (node.right != null) { queue.add(node.right); } }}
BFS使用场景:
更多参考:
void dfs(TreeNode root) { if (root == null) { return; } dfs(root.left); dfs(root.right);}
DFS同样可以实现层序遍历
每次递归的时候都需要带一个 index(表示当前的层数),也就对应那个田字格子中的第几行,如果当前行对应的 list 不存在,就加入一个空 list 进去。
动态演示如下:
参考:
import java.util.*; class Solution { public List
> levelOrder(TreeNode root) { if(root==null) { return new ArrayList
>(); } //用来存放最终结果 List
> res = new ArrayList
>(); dfs(1,root,res); return res; } void dfs(int index,TreeNode root, List
> res) { //假设res是[ [1],[2,3] ], index是3,就再插入一个空list放到res中 if(res.size() ()); } //将当前节点的值加入到res中,index代表当前层,假设index是3,节点值是99 //res是[ [1],[2,3] [4] ],加入后res就变为 [ [1],[2,3] [4,99] ] res.get(index-1).add(root.val); //递归的处理左子树,右子树,同时将层数index+1 if(root.left!=null) { dfs(index+1, root.left, res); } if(root.right!=null) { dfs(index+1, root.right, res); } }}
转载地址:http://nglr.baihongyu.com/