Binary Tree Preorder Traversal--leetcode

来源:互联网 发布:java log4j使用 编辑:程序博客网 时间:2024/06/09 23:51

原题链接:https://oj.leetcode.com/problems/binary-tree-preorder-traversal/

题目大意:先序遍历二叉树

思路:该算法属于基础算法,需要掌握其递归和非递归实现。面试时,要做到一次bug-free。递归思路和实现很简单,这里就不说了,下面主要介绍先序遍历的非递归实现算法。

我们用一个栈来模拟递归过程,若结点不空,则访问并入栈;若节点为空,但栈不空,则弹出栈顶元素,并访问其右子树。栈用于回溯。

时间复杂度:因为每个节点访问一次,所以时间复杂度为O(N)。 空间复杂度为树的深度 O(lgN)。

class Solution {public:    vector<int> preorderTraversal(TreeNode *root) {        vector<int> res;        if(!root)            return res;        stack<TreeNode *> s;        while(root||!s.empty()){            if(root){                s.push(root);                res.push_back(root->val);                root=root->left;            }            else{                root=s.top();                s.pop();                root=root->right;            }        }        return res;    }};

熟练掌握,做到bug free。树的遍历是许多与树相关的算法基础,所以应该重视。


5 0
原创粉丝点击