码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode--Binary Tree Level Order Traversal

时间:2017-06-14 18:09:09      阅读:149      评论:0      收藏:0      [点我收藏+]

标签:div   ret   vector   tree   tco   down   addclass   push   bool   

Binary Tree Level Order Traversal

这几个题目相似。都是用同样的递归法,时间空间复杂度都为O(n).

当然,还有迭代法,控件复杂度会降至 O(1),这个暂不讨论。

Binary Tree Level Order Traversal

// 递归版,时间复杂度 O(n),空间复杂度 O(n) class Solution {
public:
      vector<vector<int> > levelOrder(TreeNode *root) {
          vector<vector<int>> result;
          traverse(root, 1, result);
          return result;
}
      void traverse(TreeNode *root, size_t level, vector<vector<int>> &result) {
          if (!root) return;
          if (level > result.size())
              result.push_back(vector<int>());
          result[level-1].push_back(root->val);
          traverse(root->left, level+1, result);
          traverse(root->right, level+1, result);
} };

Binary Tree Level Order Traversal II

// 递归版,时间复杂度 O(n),空间复杂度 O(n)
class Solution {
public:
vector<vector<int> > levelOrderBottom(TreeNode *root) { vector<vector<int>> result;
traverse(root, 1, result);
std::reverse(result.begin(), result.end()); // 比上一题多此一行 return result;
}
      void traverse(TreeNode *root, size_t level, vector<vector<int>> &result) {
          if (!root) return;
          if (level > result.size())
              result.push_back(vector<int>());
          result[level-1].push_back(root->val);
          traverse(root->left, level+1, result);
          traverse(root->right, level+1, result);
} };

Binary Tree Zigzag Level Order Traversal

// 递归版,时间复杂度 O(n),空间复杂度 O(n)
class Solution {
public:
      vector<vector<int> > zigzagLevelOrder(TreeNode *root) {
          vector<vector<int>> result;
          traverse(root, 1, result, true);
          return result;
}
      void traverse(TreeNode *root, size_t level, vector<vector<int>> &result,
              bool left_to_right) {
          if (!root) return;
          if (level > result.size())
              result.push_back(vector<int>());
          if (left_to_right)
              result[level-1].push_back(root->val);
          else
              result[level-1].insert(result[level-1].begin(), root->val);
          traverse(root->left, level+1, result, !left_to_right);
          traverse(root->right, level+1, result, !left_to_right);
      }
};

LeetCode--Binary Tree Level Order Traversal

标签:div   ret   vector   tree   tco   down   addclass   push   bool   

原文地址:http://www.cnblogs.com/mthoutai/p/7009995.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!