iT邦幫忙

2024 iThome 鐵人賽

DAY 16
0
佛心分享-刷題不只是刷題

刷經典 LeetCode 題目系列 第 16

經典LeetCode 102. Binary Tree Level Order Traversal

  • 分享至 

  • xImage
  •  

這道題要求我們對一棵二元樹進行層次遍歷,也就是按照樹的層級順序,從左到右依次訪問每一層的節點,並將每層的節點值作為一個子清單存入結果中。

題目:

給定一棵二元樹的根節點 root,回傳其節點值的層次遍歷(從左到右,逐層進行)。

  • 輸入:二元樹的根節點 root
  • 輸出:每層節點值的清單,按層次順序排列。

範例:

範例 1:

輸入:
root:
     3
   /   \
  9    20
      /  \
     15   7

輸出:
[
  [3],
  [9,20],
  [15,7]
]

範例 2:

輸入:
root: []
輸出:[]

解題思路

這題感覺跟 leetcode 104 有關係,可以用上相同技巧跟套路,他是要印出每一層的節點,所以會使用廣度優先 BFS 的方式來解,在遍歷二元樹的同時放入當層的節點到 vector 裡,

實作:

/**
 * 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>> levelOrder(TreeNode* root) {
        if (root == nullptr) {
            return {};
        }

        vector<vector<int>> res;
        vector<int> level;
        queue<TreeNode*> q;
        q.push(root);

        while (!q.empty()) {
            int size = q.size(); // 記住這層有幾個節點要處理
            level.clear();
            for (int i = 0; i < size; i++) {
                TreeNode* node = q.front();
                q.pop();

                // 重點邏輯:處理當前節點
                level.push_back(node->val);

                if (node->left != nullptr) {
                    q.push(node->left);
                }
                if (node->right != nullptr) {
                    q.push(node->right);
                }
            }
            res.push_back(level);
        }
        return res;
    }
};

參考:
#102. Binary Tree Level Order Traversal


上一篇
經典LeetCode 572. Subtree of Another Tree
下一篇
經典LeetCode 230. Kth Smallest Element in a BST
系列文
刷經典 LeetCode 題目43
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言