LeetCode(102)Binary Tree Level Order Traversal

最新推荐文章于 2020-01-03 13:57:10 发布

原创最新推荐文章于 2020-01-03 13:57:10 发布 · 2.5k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #树

C++ 同时被 3 个专栏收录

262 篇文章

订阅专栏

InterviewQuestions

183 篇文章

订阅专栏

DataStructure_Algorithm

175 篇文章

订阅专栏

题目如下：

Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).

For example:
Given binary tree {3,9,20,#,#,15,7},

return its level order traversal as:

[
  [3],
  [9,20],
  [15,7]
]

分析如下：

本题考察二叉树的层序遍历，之前也写过类似的主题。本题我尝试着不在IDE中写，而是直接在Leetcode网页提供的编辑器上写。最后自认为非常小心翼翼地检查好了，在IDE中跑了跑，发现挂了，走入了死循环。于是调试之，找bug之。终于提交。整体花了60分钟左右。远远低于面试要求平均每道题25分钟搞定并且bug-free的要求，哎。哎。

//20ms过大集合
/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
#include <queue>;
using namespace std;
class Solution {
public:
    vector<vector<int> > levelOrder(TreeNode *root) {
        vector<vector<int>> res_vec_vec;
        if(root==NULL)
            return res_vec_vec;
        queue<queue<TreeNode*>> que_que;
        queue<TreeNode*> que;
        que.push(root);
        que_que.push(que);
        while(!que_que.empty()){
            queue<TreeNode*> tmp_src_queue=que_que.front();
            queue<TreeNode*> tmp_out_queue;
            vector<int> tmp_vec;
            while(!tmp_src_queue.empty()) {
                tmp_vec.push_back(tmp_src_queue.front()->val);
                TreeNode* tmp_node=tmp_src_queue.front();
                if(tmp_node->left!=NULL)
                    tmp_out_queue.push(tmp_node->left);
                if(tmp_node->right!=NULL)
                    tmp_out_queue.push(tmp_node->right);
                tmp_src_queue.pop();
            }
            res_vec_vec.push_back(tmp_vec);
            if(!tmp_out_queue.empty())//如果没有这个条件判断，程序就死循环了
                que_que.push(tmp_out_queue);
            que_que.pop();
        }
        return res_vec_vec;
    }

};

小结:

(1) bug-free很重要，现在就要开始练习，要不然面试必挂。
(2) bug-free的实际意义是，在真正的开发中，省去找bug的时间。与其你写个漏洞百出的code然后等着一点一点地被修正，花去大量时间。不如从一开始就像清楚，写清楚。
(3) 本题的一个重要bug是下面的这句话

            if(!tmp_out_queue.empty())//如果没有这个条件判断，程序就死循环了
                que_que.push(tmp_out_queue);

之前没有写if()条件就直接push了，导致死循环。这个bug的实质在于，对于循环的边界条件的判断还是做得不够好，两层循环一出来，立刻暴露缺陷。

update : 2015-03-24

受到 Subset这道题目的启发，用size记录下当前level的节点个数，这样就可以简化很多代码。

//11ms 
class Solution {
public:
    vector<vector<int> > levelOrder(TreeNode *root) {
        queue<TreeNode*> q;
        if (root != NULL) {
            q.push(root);
        }
        vector<int> inner;
        vector<vector<int> > outer;
        TreeNode* node;
        while(!q.empty()) {
            int size = q.size();
            inner.clear();
            while (size-- > 0) {
                node = q.front();
                q.pop();
                inner.push_back(node->val);
                if (node->left != NULL)
                    q.push(node->left);
                if (node->right != NULL)
                    q.push(node->right);
            }
            outer.push_back(inner);
        }
        return outer;
    }
};