您的位置:首页 > 其它

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

2015-02-23 12:52 543 查看
Problem:

I:

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}
,

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}
,

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++:

I:

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:
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: