码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode Binary Tree Zigzag Level Order Traversal

时间:2014-11-28 14:18:16      阅读:131      评论:0      收藏:0      [点我收藏+]

标签:style   blog   io   ar   color   sp   for   on   数据   

给定一个树,按照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;
}
};

 

leetcode Binary Tree Zigzag Level Order Traversal

标签:style   blog   io   ar   color   sp   for   on   数据   

原文地址:http://www.cnblogs.com/higerzhang/p/4128225.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!