力扣实战记录第144题——二叉树的前序遍历

这是我在力扣平台做的第二题，昨天复习了二叉树的遍历方法，今天打算用两种遍历方法——递归、非递归的方法去实现二叉树的遍历，先贴上我用递归的方法做的第一种解法：

/*
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> var;
    vector<int> preorderTraversal(TreeNode* root) {
        if(root!=NULL){
            var.push_back(root->val);
            preorderTraversal(root->left);
            preorderTraversal(root->right);
        }
        return var;
    }
};

递归的思路比较简单，就是先序遍历的思路，先根，再左，再右，由于不知道自己可以往这个类里添加成员变量，所以又费了一点劲，得到的收获是这个类就是自己的，可以往其中添加任意的东西。

下面贴上非递归遍历的解法：

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

解释一下，首先，沿着左子树访问左边的节点，并将这些节点保存下来，这些节点访问顺序肯定是前序遍历，直到没有左子树了，这个时候就可以访问右子树，通过拿出之前栈中保存的节点，继续访问他的左子树，如果这个节点没有左子树了，就访问他的右子树，每一次的这种访问伴随着从栈中拿出这个节点，以代表右子树已经开始访问或者访问完成。