简单易理解一个模板解决非递归前根、中根、后根遍历(leetcode144,94,145)

简单易理解一个模板解决非递归前根、中根、后根遍历(leetcode144,94,145)

• 先跟遍历总体思路：存放节点值 -> 访问左子树 -> 访问右子树。
• 中根遍历总体思路：访问左子树 -> 存放节点值 -> 访问右子树。
• 后根遍历总体思路：访问左子树 -> 访问右子树 -> 存放节点值。

三种非递归遍历有多种解法，此文将介绍一种通用模板，只需要略加修改便可同时解决三种遍历方式。

下面将把三种代码贴出来，略加修改的地方将在注释中详细指出。

一、先根遍历(leetcode 144)

详细讲解放在注释中。

// 输出一棵树的先根遍历
vector<int> preorderTraversal(TreeNode* root) {  
        vector<int> ans;   // 存放最后结果
        stack<TreeNode*> node_stack;   // 使用栈来解决此题
        TreeNode* node = root;   // 遍历用的节点，初始化将此节点指向根节点
        
    	// 开始遍历
    	// 循环条件：当树中还存在未被遍历的节点
        while (node != nullptr || !node_stack.empty()) {
            // 如果当前节点不为空，继续遍历左子树
            if (node != nullptr) {
                
                // 代码其它部分均为模板，三种遍历方式的区别只有此行放置的位置
                ans.push_back(node->val);   
                
                node_stack.push(node);  // 将此节点入栈，开始访问左子树
                node = node->left;
            }
            else {  // 若此节点为空，即到达树的叶子节点
                node = node_stack.top();  // 返回上一个节点
                node_stack.pop();
                node = node->right;   // 访问其右子树
            }
        }
        
    	// 返回结果
        return ans;
    }
};

二、中根遍历(leetcode 94)

中根遍历其余部分与先根遍历一样，只有ans.push_back(node->val);此句位置不同。

// 中根遍历只有一句话的位置和先根不同
vector<int> inorderTraversal(TreeNode* root) {
        if (root == nullptr) {
            return vector<int>();
        }
        
        TreeNode* node = root;
        stack<TreeNode*> node_stack;
        vector<int> ans;
        
        while (node != nullptr || !node_stack.empty()) {
            if (node != nullptr) {
                node_stack.push(node);
                node = node->left;  
                
            }
            else {
                node = node_stack.top();
                node_stack.pop();
                ans.push_back(node->val);    // 存放节点值，只有此处位置与先根不同
                node = node->right;   
            }
        }
        
        return ans;
    }

三、后根遍历(leetcode 145)

在后根遍历的思路中，访问顺序为：左右根。将此顺序反转：根右左。可以将后根遍历看成先访问右子树的先根遍历，将结果反转，下面即用此思路，代码实现和先根遍历几乎完全一样，只不过先遍历右子树，并且在最后反转结果。

// 代码实现和先根遍历几乎完全相同，只不过先访问右子树，结果再反转。
vector<int> postorderTraversal(TreeNode* root) {
        vector<int> ans;
        stack<TreeNode*> node_stack;
        TreeNode* node = root;
        
        while (node != nullptr || !node_stack.empty()) {
            if (node != nullptr) {
                ans.push_back(node->val);
                node_stack.push(node);  // 存放节点值
                node = node->right;   // 先访问右子树，只有访问顺序与先根不同
            }
            else {
                node = node_stack.top();
                node_stack.pop();
                node = node->left;   // 再访问左子树
            }
        }
        
        reverse(ans.begin(), ans.end());   // 反转结果
        return ans;
    }

到此，三种遍历方式结束了，中根、后根的遍历方式与先根可以说非常相似，十分便于记忆。