代码随想录第三十四天

二叉树-递归遍历与迭代遍历

递归三要素

1. 确定递归函数的参数和返回值： 确定哪些参数是递归的过程中需要处理的，那么就在递归函数里加上这个参数， 并且还要明确每次递归的返回值是什么进而确定递归函数的返回类型。
2. 确定终止条件： 写完了递归算法, 运行的时候，经常会遇到栈溢出的错误，就是没写终止条件或者终止条件写的不对，操作系统也是用一个栈的结构来保存每一层递归的信息，如果递归没有终止，操作系统的内存栈必然就会溢出。
3. 确定单层递归的逻辑： 确定每一层递归需要处理的信息。在这里也就会重复调用自己来实现递归的过程。

class Solution {
public:
    void traversal(TreeNode* cur, vector<int>& vec){
        
        if(cur == NULL)return;
        //前序
        vec.push_back(cur->val);
        traversal(cur->left, vec);
        traversal(cur->right, vec);
        //中序
        //traversal(cur->left, vec);  // 左
        //vec.push_back(cur->val);    // 中
        //traversal(cur->right, vec); // 右
    	
        //后序
        //traversal(cur->left, vec);  // 左
        //traversal(cur->right, vec); // 右
    	//vec.push_back(cur->val);    // 中
    }
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int>res;
        traversal(root,res);
        return res;
    }
};

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        stack<TreeNode*> st;
        vector<int> res;
        if(root == NULL) return res;
        st.push(root);
        while(!st.empty()){
            TreeNode* node = st.top();
            st.pop();
            res.push_back(node->val);
            if (node->right) st.push(node->right);//如果是后序遍历就将这两行调换
            if (node->left) st.push(node->left);
        }
        //后序遍历还需要将结果反转，就是左中右的顺序了
        //reverse(res.begin(), res.end());
         return res;
    }
};

迭代的中序遍历需要用到指针

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

        }
        return res;
    }
};