二叉树的层次遍历

来源:互联网 发布:网络写手水军做的坏事 编辑:程序博客网 时间:2024/05/22 15:38

描述:给出一棵二叉树,返回其节点值的层次遍历(逐层从左往右访问)


样例:给一棵二叉树 {3,9,20,#,#,15,7} :


  3
 / \
9  20
  /  \
 15   7
返回他的分层遍历结果:


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

解题思路:这个题与普通的层序遍历之间的区别在于,要把每一层的节点分别输出。我们可以把每一层的节点都放到一个vector向量里面,然后再把每一层的vector都放入到一个总的vector里面。

实现代码:

/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */
 
  class Solution {
    /**
     * @param root: The root of binary tree.
     * @return: Level order a list of lists of integer
     */
    public: vector<vector<int>> levelOrder(TreeNode *root) {
        // write your code here
    vector<vector<int>> result;
        if (root == NULL) {
            return result;
        }
        
        queue<TreeNode *> Q;
        Q.push(root);
        while (!Q.empty()) {
            int size = Q.size();
            vector<int> level;
            
            for (int i = 0; i < size; i++) {
                TreeNode *head = Q.front(); Q.pop();
                level.push_back(head->val);
                if (head->left != NULL) {
                    Q.push(head->left);
                }
                if (head->right != NULL) {
                    Q.push(head->right);
                }
            }
            
            result.push_back(level);
        }
        
        return result;
    }
};

做题感想:首先会用到队列的知识,对于每一个节点首先让它入队然后再出队。其次怎样把每一层的节点放入到vector当中,当时写的时候出来好多错误,不能正确的判断,怎样到一层的结束。后来费了老鼻子劲才写对。

0 0