Leetcode 103. 二叉树的锯齿形层次遍历 解题思路及C++实现
生活随笔
收集整理的這篇文章主要介紹了
Leetcode 103. 二叉树的锯齿形层次遍历 解题思路及C++实现
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
解題思路:
因為是逐層的訪問,在訪問某一層的節點時,需要將其子節點push進棧,所以自然而然地想到用兩個棧a和b。因為是鋸齒形的訪問順序,所以,如果將根節點當做第一層,奇數層與偶數層的進棧區別在于:奇數層節點的左子節點先進棧,右子節點后進棧,訪問偶數層節點時,順序相反,詳細可看下面的代碼。下面的程序中,使用了一個flag來判斷是奇數層還是偶數層,沒循環一層,flag做一次正負反轉。
大循環的邏輯是逐層訪問,每訪問一層,棧a和棧b,必有一個是空的。循環停止條件是兩個棧都為空。
?
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode(int x) : val(x), left(NULL), right(NULL) {}* };*/ class Solution { public:vector<vector<int>> zigzagLevelOrder(TreeNode* root) {if(root == NULL) return {};vector<vector<int> > res;int flag = 1;stack<TreeNode*> a, b;a.push(root);while(!a.empty() || !b.empty()){vector<int> cur; //存儲每一層的訪問結果while(flag == 1 && !a.empty()){ //奇數層(把根節點當做第1層)root = a.top();cur.push_back(root->val);a.pop();if(root->left != NULL) b.push(root->left);if(root->right != NULL) b.push(root->right);}while(flag == -1 && !b.empty()){ //偶數層root = b.top();cur.push_back(root->val);b.pop();if(root->right != NULL) a.push(root->right);if(root->left != NULL) a.push(root->left);}flag = -1 * flag; //每訪問一層,反轉一下res.push_back(cur); //存到結果容器中}return res;} };?
?
?
?
?
?
總結
以上是生活随笔為你收集整理的Leetcode 103. 二叉树的锯齿形层次遍历 解题思路及C++实现的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Leetcode 94. 二叉树的中序遍
- 下一篇: Leetcode 144. 二叉树的前序