[Leetcode 102 and 107, Easy] Binary Tree Level Order Traversal (I and II)

最新推荐文章于 2020-12-15 02:41:07 发布

simplify2012

最新推荐文章于 2020-12-15 02:41:07 发布

阅读量420

点赞数

分类专栏： Leetcode_easy 文章标签： leetcode Binary Tree

本文链接：https://blog.csdn.net/u012212811/article/details/43916003

版权

Leetcode_easy 专栏收录该内容

58 篇文章 0 订阅

订阅专栏

Problem:

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]
]

II:

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

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

return its bottom-up level order traversal as:

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

Analysis:

The below solutions are recursive methods. The main difference between the below solutions and the typical in-order traseversal is we use an int variable to flag the current level in the process of recursion.

In the second solution, we use the odd-even property of this variable to determine how to add val to the vectors, push_back or insert.

Solutions:

C++:

vector<vector<int> > levelOrder(TreeNode *root) {
        vector<vector<int> > keys;
        
        if(root == NULL)
            return keys;
        
        queue<TreeNode*> nodes;
        vector<int> level;
        level.push_back(root->val);
        keys.push_back(level);
        level.clear();
        nodes.push(root);
        nodes.push(NULL);
        
        while(!nodes.empty()) {
            if(nodes.front() == NULL) {
                if(!level.empty())
                    keys.push_back(level);
                level.clear();
                nodes.pop();
                continue;
            }
            
            TreeNode* local_root = nodes.front();
            nodes.pop();
            if(local_root->left) {
                level.push_back(local_root->left->val);
                nodes.push(local_root->left);
            }
            
            if(local_root->right) {
                level.push_back(local_root->right->val);
                nodes.push(local_root->right);
            }
            
            if(nodes.front() == NULL)
                nodes.push(NULL);
        }
        
        return keys;
    }

II:

vector<vector<int> > levelOrderBottom(TreeNode *root) {
        vector<vector<int> > keys;
        stack<vector<int> > stack_keys;
        
        if(root == NULL)
            return keys;
        
        queue<TreeNode*> nodes;
        vector<int> level;
        level.push_back(root->val);
        stack_keys.push(level);
        level.clear();
        nodes.push(root);
        nodes.push(NULL);
        
        while(!nodes.empty()) {
            if(nodes.front() == NULL) {
                if(!level.empty())
                    stack_keys.push(level);
                level.clear();
                nodes.pop();
                continue;
            }
            
            TreeNode* local_root = nodes.front();
            nodes.pop();
            if(local_root->left) {
                level.push_back(local_root->left->val);
                nodes.push(local_root->left);
            }
            
            if(local_root->right) {
                level.push_back(local_root->right->val);
                nodes.push(local_root->right);
            }
            
            if(nodes.front() == NULL)
                nodes.push(NULL);
        }
        
        while(!stack_keys.empty()) {
            keys.push_back(stack_keys.top());
            stack_keys.pop();
        }
        
        return keys;
        
    }

Java :

Python: