1. 題目

給定一個二叉樹，編寫一個函數來獲取這個樹的最大寬度。樹的寬度是所有層中的最大寬度。這個二叉樹與滿二叉樹（full binary tree）結構相同，但一些節點為空。

每一層的寬度被定義為兩個端點（該層最左和最右的非空節點，兩端點間的null節點也計入長度）之間的長度。

示例 1: 輸入: 1/ \3 2/ \ \ 5 3 9 輸出: 4 解釋: 最大值出現在樹的第 3 層，寬度為 4 (5,3,null,9)。示例 2: 輸入: 1/ 3 / \ 5 3 輸出: 2 解釋: 最大值出現在樹的第 3 層，寬度為 2 (5,3)。示例 3: 輸入: 1/ \3 2 / 5 輸出: 2 解釋: 最大值出現在樹的第 2 層，寬度為 2 (3,2)。示例 4: 輸入: 1/ \3 2/ \ 5 9 / \6 7 輸出: 8 解釋: 最大值出現在樹的第 4 層， 寬度為 8 (6,null,null,null,null,null,null,7)。 注意: 答案在32位有符號整數的表示范圍內。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/maximum-width-of-binary-tree
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

• 記錄深度，和每一層的最左側節點位置（中序遍歷每層第一次碰到的）
• 節點位置用樹的數組實現方式（注意溢出，每層pos*2很快就溢出了，用ulong能過）

class Solution {int ans; public:int widthOfBinaryTree(TreeNode* root) {if(!root) return 0;vector<unsigned long> LeftPosOflv;ans = 1;dfs(root, 0, 0, LeftPosOflv);return ans;}void dfs(TreeNode* root, int depth, unsigned long pos, vector<unsigned long> &LeftPosOflv){if(!root) return;if(LeftPosOflv.size() <= depth)//第一次遇到的是該層最左邊的LeftPosOflv.push_back(pos);ans = max(ans, int(pos-LeftPosOflv[depth]+1));dfs(root->left, depth+1, 2*pos+1, LeftPosOflv);dfs(root->right, depth+1, 2*pos+2, LeftPosOflv);} };

12 ms 16.3 MB

總結

以上是生活随笔為你收集整理的LeetCode 662. 二叉树最大宽度（递归）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。