leetcode 日经贴，Cpp code -Binary Tree Level Order Traversal II

原创于 2015-05-05 18:56:21 发布 · 340 阅读

0 ·

CC 4.0 BY-SA版权

算法同时被 3 个专栏收录

183 篇文章

订阅专栏

leetcode

173 篇文章

订阅专栏

Cpp

152 篇文章

订阅专栏

本文介绍了一种实现二叉树层次遍历的方法，并将结果以从底部开始的顺序返回。通过使用队列实现节点的逐层遍历，再反转结果顺序，达到题目要求。代码采用C++实现。

Binary Tree Level Order Traversal II

/**
 * 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>> levelOrderBottom(TreeNode* root) {
        queue<TreeNode *> q;
        vector<vector<int> >ans;
        vector<int> vi;
        if (!root) {
            return ans;
        }
        q.push(root);
        q.push(NULL);
        while (!q.empty()) {
            TreeNode *r = q.front();
            q.pop();
            if (!r) {
                ans.push_back(vi);
                vi.clear();
                if (!q.empty()) {
                    q.push(NULL);
                }
            } else {
                vi.push_back(r->val);
                if (r->left) {
                    q.push(r->left);
                }
                if (r->right) {
                    q.push(r->right);
                }
            }
        }
        reverse(ans.begin(), ans.end());
        return ans;
    }
};