牛客題霸 [二叉樹的之字形層序遍歷] C++題解/答案

題目描述

給定一個二叉樹，返回該二叉樹的之字形層序遍歷，（第一層從左向右，下一層從右向左，一直這樣交替）
例如：
給定的二叉樹是{3,9,20,#,#,15,7},

該二叉樹之字形層序遍歷的結果是
[
[3],
[20,9],
[15,7]
]

題解：

因為是之字形，所以如果是偶數行就從左向右遍歷，如果是奇數行就從右向左遍歷，我們可以用一個bool變量flag來標記當前是偶還是奇

代碼：

/*** struct TreeNode {* int val;* struct TreeNode *left;* struct TreeNode *right;* };*/class Solution { public:/*** * @param root TreeNode類 * @return int整型vector<vector<>>*/vector<vector<int> > zigzagLevelOrder(TreeNode* root) {// write code hereif(root == nullptr) return {};vector<vector<int>> res;queue<TreeNode*> q;q.push(root);vector<int> temp;bool flag = 0;while(q.size()){int n = q.size();while(n--){TreeNode* node = q.front();q.pop();temp.push_back(node->val);if(node->left) q.push(node->left);if(node->right) q.push(node->right);}if(flag) reverse(temp.begin(),temp.end());//翻轉res.push_back(temp);temp.clear();flag = !flag;}return res;} };

總結

以上是生活随笔為你收集整理的牛客题霸 [二叉树的之字形层序遍历] C++题解/答案的全部內容，希望文章能夠幫你解決所遇到的問題。

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