目录
⼆叉树的锯⻮形层序遍历(medium)
题目解析
讲解算法原理
编写代码
⼆叉树的最⼤宽度(medium)
题目解析
讲解算法原理
编写代码
⼆叉树的锯⻮形层序遍历(medium)
题目解析
1.题目链接:. - 力扣(LeetCode)
2.题目描述
给你⼆叉树的根节点root,返回其节点值的锯⻮形层序遍历。(即先从左往右,再从右往左进⾏下⼀层遍历,以此类推,层与层之间交替进⾏)。
⽰例1:输⼊:root=[3,9,20,null,null,15,7]
输出:[[3],[20,9],[15,7]]
⽰例2:
输⼊:root=[1]
输出:[[1]]
⽰例3:
输⼊:root=[]
输出:[]
讲解算法原理
解法(层序遍历):
算法思路:
在正常的层序遍历过程中,我们是可以把⼀层的结点放在⼀个数组中去的。
既然我们有这个数组,在合适的层数逆序就可以得到锯⻮形层序遍历的结果。
编写代码
c++算法代码:
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode() : val(0), left(nullptr), right(nullptr) {}* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left),
right(right) {}* };*/
class Solution
{
public:vector<vector<int>> zigzagLevelOrder(TreeNode* root) {vector<vector<int>> ret;if(root == nullptr) return ret;queue<TreeNode*> q;q.push(root);int level = 1;while(q.size()){int sz = q.size();vector<int> tmp;for(int i = 0; i < sz; i++){auto t = q.front();q.pop();tmp.push_back(t->val);if(t->left) q.push(t->left);if(t->right) q.push(t->right);}// 判断是否逆序if(level % 2 == 0) reverse(tmp.begin(), tmp.end());ret.push_back(tmp);level++;}return ret;}
};
java算法代码:
/*** Definition for a binary tree node.* public class TreeNode {* int val;* TreeNode left;* TreeNode right;* TreeNode() {}* TreeNode(int val) { this.val = val; }* TreeNode(int val, TreeNode left, TreeNode right) {* this.val = val;* this.left = left;* this.right = right;* }* }*/
class Solution
{public List<List<Integer>> zigzagLevelOrder(TreeNode root) {List<List<Integer>> ret = new ArrayList<>();if(root == null) return ret;Queue<TreeNode> q = new LinkedList<>();q.add(root);int level = 1;while(!q.isEmpty()){int sz = q.size();List<Integer> tmp = new ArrayList<>();for(int i = 0; i < sz; i++){TreeNode t = q.poll();tmp.add(t.val);if(t.left != null) q.add(t.left);if(t.right != null) q.add(t.right);}// 判断是否逆序if(level % 2 == 0) Collections.reverse(tmp);ret.add(tmp);level++;}return ret;}
}
⼆叉树的最⼤宽度(medium)
题目解析
1.题目链接:. - 力扣(LeetCode)
2.题目描述
给你⼀棵⼆叉树的根节点root,返回树的最⼤宽度。树的最⼤宽度是所有层中最⼤的宽度。
每⼀层的宽度被定义为该层最左和最右的⾮空节点(即,两个端点)之间的⻓度。将这个⼆叉树视作与满⼆叉树结构相同,两端点间会出现⼀些延伸到这⼀层的null节点,这些null节点也计⼊⻓度。
题⽬数据保证答案将会在32位带符号整数范围内。⽰例1:输⼊:root=[1,3,2,5,3,null,9]输出:4
解释:
最⼤宽度出现在树的第3层,宽度为4(5,3,null,9)。⽰例2:输⼊:root=[1,3,2,5,null,null,9,6,null,7]输出:7
解释:
最⼤宽度出现在树的第4层,宽度为7(6,null,null,null,null,null,7)。
讲解算法原理
解法(层序遍历):
算法思路:
1. 第⼀种思路(会超过内存限制):既然统计每⼀层的最⼤宽度,我们优先想到的就是利⽤层序遍历,把当前层的结点全部存在队列⾥
⾯,利⽤队列的⻓度来计算每⼀层的宽度,统计出最⼤的宽度。但是,由于空节点也是需要计算在内的。因此,我们可以选择将空节点也存在队列⾥⾯。
这个思路是我们正常会想到的思路,但是极端境况下,最左边⼀条⻓链,最右边⼀条⻓链,我们需要存⼏亿个空节点,会超过最⼤内存限制。
2. 第⼆种思路(利⽤⼆叉树的顺序存储-通过根节点的下标,计算左右孩⼦的下标):依旧是利⽤层序遍历,但是这⼀次队列⾥⾯不单单存结点信息,并且还存储当前结点如果在数组中存
储所对应的下标(在我们学习数据结构-堆的时候,计算左右孩⼦的⽅式)。
这样我们计算每⼀层宽度的时候,⽆需考虑空节点,只需将当层结点的左右结点的下标相减再加 1 即可。
但是,这⾥有个细节问题:如果⼆叉树的层数⾮常恐怖的话,我们任何⼀种数据类型都不能存下下标的值。但是没有问题,因为
• 我们数据的存储是⼀个环形的结构;
• 并且题⽬说明,数据的范围在 int 这个类型的最⼤值的范围之内,因此不会超出⼀圈;
• 因此,如果是求差值的话,我们⽆需考虑溢出的情况。
编写代码
c++算法代码:
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode() : val(0), left(nullptr), right(nullptr) {}* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left),
right(right) {}* };*/
class Solution
{
public:int widthOfBinaryTree(TreeNode* root) {vector<pair<TreeNode*, unsigned int>> q; // ⽤数组模拟队列q.push_back({root, 1});unsigned int ret = 0;while(q.size()){// 先更新这⼀层的宽度auto& [x1, y1] = q[0];auto& [x2, y2] = q.back();ret = max(ret, y2 - y1 + 1);// 让下⼀层进队vector<pair<TreeNode*, unsigned int>> tmp; // 让下⼀层进⼊这个队列 for(auto& [x, y] : q){if(x->left) tmp.push_back({x->left, y * 2});if(x->right) tmp.push_back({x->right, y * 2 + 1});}q = tmp;}return ret;}
};
java算法代码:
/*** Definition for a binary tree node.* public class TreeNode {* int val;* TreeNode left;* TreeNode right;* TreeNode() {}* TreeNode(int val) { this.val = val; }* TreeNode(int val, TreeNode left, TreeNode right) {* this.val = val;* this.left = left;* this.right = right;* }* }*/
class Solution
{public int widthOfBinaryTree(TreeNode root) {List<Pair<TreeNode, Integer>> q = new ArrayList<>(); // ⽤数组模拟队列 q.add(new Pair<TreeNode, Integer>(root, 1));int ret = 0; // 记录最终结果while(!q.isEmpty()){// 先更新⼀下这⼀层的宽度Pair<TreeNode, Integer> t1 = q.get(0);Pair<TreeNode, Integer> t2 = q.get(q.size() - 1);ret = Math.max(ret, t2.getValue() - t1.getValue() + 1);// 让下⼀层进队List<Pair<TreeNode, Integer>> tmp = new ArrayList<>();for(Pair<TreeNode, Integer> t : q){TreeNode node = t.getKey();int index = t.getValue();if(node.left != null){tmp.add(new Pair<TreeNode, Integer>(node.left, index * 2));}if(node.right != null){tmp.add(new Pair<TreeNode, Integer>(node.right, index * 2
+ 1));}}q = tmp;}return ret;}
}