给定一个树,按照Z字形记录每一行。例如:
Given binary tree {3,9,20,#,#,15,7}
,
3 / \ 9 20 / \ 15 7
return its zigzag level order traversal as:
[ [3], [20,9], [15,7] ]
这题想了有一会,按照上一题思路的启发。这题也是应该要用某个数据结构会比较好解答。然后就想用队列,然后可以任意从前面删除就好了,可以队列之提供后面pop的操作。于是就想到用栈。
思路:用两个栈,每个栈存一层,例如上一题,第一个栈sta1,先push(3),然后去sta1的元素左边右边先后入sta2,那么再pop就是20,9了,当然pop的时候又要对其进行push栈sta1,先把7push然后15push,最后栈sta1 pop的时候就是15,7了。
/** * Definition for binary tree * 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) { vector<vector<int> > ans; if (!root) return ans; vector<int> tmp; stack<TreeNode *> sta1, sta2; TreeNode *p; sta1.push(root); while(!sta1.empty() || !sta2.empty()) { while (!sta1.empty()) { p = sta1.top(); sta1.pop(); tmp.push_back(p -> val); if (p -> left) sta2.push(p -> left); if (p -> right) sta2.push(p -> right); } ans.push_back(tmp); tmp.clear(); if (sta1.empty() && sta2.empty()) return ans; while(!sta2.empty()) { p = sta2.top(); sta2.pop(); tmp.push_back(p -> val); if (p -> right) sta1.push(p -> right); if (p -> left) sta1.push(p -> left); } ans.push_back(tmp); tmp.clear(); } return ans; } };