LeetCode102. Binary Tree Level Order Traversal（C++）

原创于 2019-02-13 11:15:11 发布 · 252 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

110 篇文章

订阅专栏

本文介绍了一种解决二叉树层次遍历问题的方法，通过深度优先搜索(DFS)实现节点值从左到右，按层返回。示例中使用了一个具体的二叉树结构，并详细解释了如何通过递归调用DFS函数来获取各层节点的值。

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,null,null,15,7],

return its level order traversal as:

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

解题思路：理论上应该用BFS进行层次遍历，不过感觉链式用DFS好像更方便(＾－＾)V

/**
 * 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>>result;
    int maxlevel=0;
    vector<vector<int>> levelOrder(TreeNode* root) {
        if(root==NULL)
            result.resize(0);
        else{
            result.resize(1005);
            dfs(root,0);
            result.resize(maxlevel+1);
        }  
        return result;
    }
    void dfs(TreeNode *root, int level){
        result[level].push_back(root->val);
        maxlevel=max(maxlevel,level);
        if(root->left!=NULL)
            dfs(root->left,level+1);
        if(root->right!=NULL)
            dfs(root->right,level+1);
    }
};