力扣：145. 二叉树的后序遍历

题目描述

给你一棵二叉树的根节点root，返回其节点值的后序遍历。

问题分析

递归法： 后序遍历是左$\to$右$\to$中的顺序，所以递归逻辑是先处理左子树和右子树，再取中节点的数值。
迭代法： 后序遍历的顺序是左$\to$右$\to$中，只需要调整前序遍历的代码顺序，变成中$\to$右$\to$左的遍历顺序，然后反转result数组，输出结果的顺序就是左$\to$右$\to$中了。
二叉树的前序遍历
二叉树的中序遍历

代码实现

// 编程软件：VS2019
// 参考书籍：代码随想录
#include<iostream>
#include<vector>
#include<stack>
using namespace std;

// 定义二叉树
struct TreeNode {
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode(int v):val(v),left(nullptr),right(nullptr){}
};

// 递归法
void traversal(TreeNode* cur, vector<int>& vec) {
	if (cur == NULL) return;
	traversal(cur->left, vec);  // 递归左孩子 
	traversal(cur->right, vec); // 递归右孩子 
	vec.push_back(cur->val);    // 当前节点
}
vector<int> postorderTraversal1(TreeNode* root) {
	vector<int> result;
	traversal(root, result);
	return result;
}

// 迭代法：前序遍历是中->左->右 ———>（调整代码顺序） 中->右->左 ———> (反转result数组）左->右->中
vector<int> postorderTraversal2(TreeNode* root) {
	stack<TreeNode*> st;
	vector<int> result;
	if (root == NULL) return result;
	st.push(root);
	while (!st.empty()) {
		TreeNode* node = st.top();
		st.pop();
		result.push_back(node->val);
		// 相对于前序遍历，这里更改入栈的顺序（空节点不入栈）
		if (node->left) st.push(node->left);
		if (node->right) st.push(node->right);
	}
	// 将结果反转之后就是左->右->中的顺序了
	reverse(result.begin(), result.end());
	return result;
}

// 输出数组v中元素的值 
void print(vector<int> v) {
	for (int i = 0; i < v.size(); i++) {
		cout << v[i] << " ";
	}
	cout << endl;
}

int main() {
	TreeNode* root = new TreeNode(1);
	root->left = new TreeNode(2);
	root->right = new TreeNode(3);
	root->left->left = new TreeNode(4);
	root->left->right = new TreeNode(5);
	root->left->left->left = new TreeNode(6);
	vector<int> res;
	res = postorderTraversal2(root);
	print(res);
}
// 结果：6 4 5 2 3 1