您好,登錄后才能下訂單哦!
本篇文章給大家分享的是有關如何進行python二叉樹的層次遍歷,小編覺得挺實用的,因此分享給大家學習,希望大家閱讀完這篇文章后可以有所收獲,話不多說,跟著小編一起來看看吧。
給定一個二叉樹,返回其按層次遍歷的節點值。(即逐層地,從左到右訪問所有節點)。
例如:
給定二叉樹: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其層次遍歷結果:
[
[3],
[9,20],
[15,7]
]
答案:
1public List<List<Integer>> levelOrder1(TreeNode root) {
2 Queue<TreeNode> queue = new LinkedList<TreeNode>();
3 List<List<Integer>> wrapList = new LinkedList<List<Integer>>();
4 if (root == null)
5 return wrapList;
6 queue.offer(root);
7 while (!queue.isEmpty()) {
8 int levelNum = queue.size();
9 List<Integer> subList = new LinkedList<Integer>();
10 for (int i = 0; i < levelNum; i++) {
11 if (queue.peek().left != null)
12 queue.offer(queue.peek().left);
13 if (queue.peek().right != null)
14 queue.offer(queue.peek().right);
15 subList.add(queue.poll().val);
16 }
17 wrapList.add(subList);
18 }
19 return wrapList;
20}
解析:
LinkedList是個鏈表,先進先出,levelNum是每一層的節點數量,一層一層的遍歷,代碼沒什么難度,下面再來看一種遞歸的寫法
1public List<List<Integer>> levelOrder(TreeNode root) {
2 List<List<Integer>> res = new ArrayList<List<Integer>>();
3 levelHelper(res, root, 0);
4 return res;
5}
6
7public void levelHelper(List<List<Integer>> res, TreeNode root, int height) {
8 if (root == null) return;
9 if (height >= res.size()) {
10 res.add(new LinkedList<Integer>());
11 }
12 res.get(height).add(root.val);
13 levelHelper(res, root.left, height + 1);
14 levelHelper(res, root.right, height + 1);
15}
height其實相當于層級,height>=res.size()說明已經到了下一層了,所以要新建一個list。
以上就是如何進行python二叉樹的層次遍歷,小編相信有部分知識點可能是我們日常工作會見到或用到的。希望你能通過這篇文章學到更多知識。更多詳情敬請關注億速云行業資訊頻道。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。