103. 二叉樹的鋸齒形層次遍歷-中等難度
阿新 • • 發佈:2020-07-11
問題描述
給定一個二叉樹,返回其節點值的鋸齒形層次遍歷。(即先從左往右,再從右往左進行下一層遍歷,以此類推,層與層之間交替進行)。
例如:
給定二叉樹[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回鋸齒形層次遍歷如下:
[
[3],
[20,9],
[15,7]
]
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
解答
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * }*/ /* dfs遍歷二叉樹,從左至右。每個元素都有個deep(深度),只需將root.val放入result.get(deep)陣列中即可。 時間複雜度O(n),擊敗97%使用者。 */ class Solution { public static List<List<Integer>> result; public static void dfs(TreeNode root,int deep){ if(root.left == null && root.right == null)return; if(root.left!=null){ if(deep+1 > (result.size())-1)result.add(new LinkedList<Integer>()); if(deep%2!=0)result.get(deep+1).add(root.left.val); else result.get(deep+1).add(0,root.left.val); dfs(root.left,deep+1); } if(root.right!=null){if(deep+1 > (result.size())-1)result.add(new LinkedList<Integer>()); if(deep%2!=0)result.get(deep+1).add(root.right.val); else result.get(deep+1).add(0,root.right.val); dfs(root.right,deep+1); } } public List<List<Integer>> zigzagLevelOrder(TreeNode root) { result = new ArrayList<List<Integer>>(); if(root == null)return result; result.add(new LinkedList<Integer>()); result.get(0).add(root.val); dfs(root,0); return result; } }