二叉树的层次遍历

问题描述

给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果:

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

思路

第一种方法:dfs
deep+1>ret.size(),在新的一层插入之前,ret.push_back(vector<int> ())都会申请一个新的vector<int>且为空。
利用deep深度,来控制每一层的元素在一个vector中。

题解

class Solution {
public:
    vector<vector<int>> ret;
    void dfs(int deep,TreeNode *cur){
        if(cur== nullptr) return;
        if(deep+1>ret.size()) ret.push_back(vector<int> ());
        ret[deep].push_back(cur->val);
        if(cur->left!= nullptr) dfs(deep+1,cur->left);
        if(cur->right!= nullptr) dfs(deep+1,cur->right);
    }
    vector<vector<int>> levelOrder(TreeNode *root) {
        dfs(0, root);
        return ret;
    }
};

思路

第二种方法:层次遍历
在访问过程中将同一层中的节点同时入队列。在将该queue中所有元素出队列的同时,将下一层的元素进队列,完成交接。

题解

class Solution {
public:
    vector<vector<int>> ret;
    vector<vector<int>> levelOrder(TreeNode *root) {
        if(root== nullptr) return ret;
        queue<TreeNode *> Q;
        Q.push(root);
        TreeNode* p;
        while (!Q.empty()){
            vector<int>a;
            int len=Q.size();
            for(int i=0;i<len;i++){
                p=Q.front();
                a.push_back(p->val);
                Q.pop();
                if(p->left!= nullptr) Q.push(p->left);
                if(p->right!= nullptr) Q.push(p->right);
            }
            ret.push_back(a);
        }
        return ret;
    }
};

标签: none

添加新评论