leetcode笔记:Binary Tree Level Order Traversal

来源:互联网 发布:c语言如何判断素数 编辑:程序博客网 时间:2024/05/15 04:59

一. 题目描述

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

二. 题目分析

无。

三. 示例代码

参考了Discussion中stellari的做法,递归进行层次遍历,并将每个level对应于相应的vector:

#include <iostream>#include <vector>using namespace std;struct TreeNode{    int val;    TreeNode *left;    TreeNode *right;    TreeNode(int x) :val(x), left(NULL), right(NULL){}};class Solution {public:    vector<vector<int> > levelOrder(TreeNode *root)    {        vector<vector<int> > result;        orderTraversal(root, 1, result);        return result;    }private:    void orderTraversal(TreeNode* root, int level, vector<vector<int> > & result)    {        if (root == NULL) return;        if (level > result.size()) // 每往下一级,创建一个空的vector保存该层数据            result.push_back(vector<int>());        result[level - 1].push_back(root->val);        orderTraversal(root->left, level + 1, result);        orderTraversal(root->right, level + 1, result);    }};

四. 小结

以上为递归实现,也可以写成迭代版本,时间复杂度O(n),空间复杂度O(1)。

3 0
原创粉丝点击