[Leetcode] Binary tree preorder traversal

Given a binary tree, return the preorder traversal of its nodes' values.

For example:
Given binary tree{1,#,2,3},

   1
    \
     2
    /
   3

 

return[1,2,3].

Note: Recursive solution is trivial, could you do it iteratively?

 

二叉树的遍历分为：先序遍历、中序遍历、后续遍历，三者的分类可以根据根节点被访问的先后顺序来区分。

先序遍历的遍历顺序是根节点->左孩子->右孩子。思路：先将根节点存入栈中，然后首先访问根节点的值，接下来在root沿着左孩子访问（更新根节点为root->left）的同时，将对应的右孩子依次存入栈中，直到root到达最左端的左孩子（root->left不存在），一旦左孩子不存在，就从栈中取出结点访问，利用栈的先进后出的特性，访问右孩子（即root=S.top()），然后重复上述过程，实现先序遍历。可以举例证明，整棵树的根结点最后出栈。代码如下：

 

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> preorderTraversal(TreeNode *root) 
    {
        vector<int> res;
        if(root==nullptr)    return res;
        stack<TreeNode*> S;
        S.push(root);
        while( !S.empty())
        {
            res.push_back(root->val);
            if(root->right)      //避免右孩子为空
                S.push(root->right);
            if(root->left)
                root=root->left;
            else
            {
                root=S.top();
                S.pop();
            }

        }

        return res;

    }
};

 

虽然题目要求不能用递归版的，但是写在这方便以后查看：

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> preorderTraversal(TreeNode *root) 
    {
        vector<int> res;
        RecursiveTra(root,res);
        return res;
    }
    void RecursiveTra(TreeNode *rot,vector<int> &res)
    {
        if(root==NULL)  return;
        res.push_back(root->val);
        RecursiveTra(root->left,res);
        RecursiveTra(root->right,res);
    }
};