LeetCode:144.二叉树的前序遍历

144.二叉树的前序遍历
二叉树的前序遍历：按照访问根节点——左子树——右子树的方式遍历这棵树，而在访问左子树或者右子树的时候，我们按照同样的方式遍历，直到遍历完整棵树。因此整个遍历过程天然具有递归的性质。

方法一：递归

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void preorder(TreeNode* root,vector<int> &res)
    {
        if(root==nullptr)
          return;
        res.push_back(root->val);
        preorder(root->left，res);
        preorder(root->right,res);
    }
    vector<int> preorderTraversal(TreeNode* root) {
       vector<int> ans;
       preorder(root,ans);
       return ans;
    }
};

方法二：迭代

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
         vector<int> ans; //定义一个空的数组
         stack<TreeNode*> s; //定义一个空的栈
         s.push(root); //栈中存入根节点
        //根左右
        //栈：先进后出
        while(!s.empty())
        {
            //存根,移除栈中取出的元素
            TreeNode* r=s.top();
            s.pop();
            if(!r)  continue;
            ans.push_back(r->val);  //将一个新的元素加到vector的最后面，位置为当前元素的下一个元素
            s.push(r->right);   //在栈顶添加元素
            s.push(r->left);
        }
        return ans;
    }
};