代码随想录算法训练营Day14|理论基础● 递归遍历 ● 迭代遍历● 统一迭代

从今天开始正式进入到二叉树章节，因为今天的内容我在学校就已经全部学过了，所以今天就是简单的说一下，题目就不放上来了，都是简单的二叉树的前中后序遍历的递归写法和迭代写法；

递归写法不用多说，来看迭代的具体代码

 vector<int> preorderTraversal(TreeNode* root) {
        if(root==NULL) return {};
        vector<int> vec;
	stack<TreeNode*> st;
	st.push(root);
	while (st.empty() == false) {
		TreeNode* node = st.top();
		st.pop();
        if(node!=NULL)
		vec.push_back(node->val);
		if (node->right != NULL) {
			st.push(node->right);
		}
		if (node->left != NULL) {
			st.push(node->left);
		}
	}
	return vec;
    }

这是前序遍历的迭代法的写法，注意这里是先将右子树加入到栈中，再将左子树加入到栈中，因为这样才能先弹出左节点，再弹出右节点

 vector<int> postorderTraversal(TreeNode* root) {
        if(root==NULL) return {};
        vector<int> vec;
	stack<TreeNode*> st;
	st.push(root);
	while (st.empty() == false) {
		TreeNode* node = st.top();
		st.pop();
		vec.push_back(node->val);
		if (node->left != NULL) {
			st.push(node->left);
		}
		if (node->right != NULL) {
			st.push(node->right);
		}
	}
	for (int i = 0; i < vec.size()/2; i++) {
		swap(vec[i], vec[vec.size() - i - 1]);
	}
	return vec;
	
    }

这是后序遍历的迭代法，这里值得注意的是，将左右结点进栈的顺序改变之后，他的顺序就是中右左，再将数组翻转过来就是左右中了

vector<int> inorderTraversal(TreeNode* root) {
       vector<int> vec;
	stack<TreeNode*> st;
	TreeNode* node = root;
	while (st.empty() == false || node != NULL) {
		if (node != NULL) {
			st.push(node);
			node = node->left;
		}
		else {
			node = st.top();
			st.pop();
			vec.push_back(node->val);
			node = node->right;
		}
	}
	return vec;
    }

这是中序遍历的迭代法，这里就是再该节点不为空的时候，不断将其加入到栈中，直到他为空的时候，代表此时我的左节点已经遍历完了，就将栈顶元素弹出，然后加入到数组中，再将node赋为该节点的右孩子，因为是中序遍历，遍历顺序是左中右，然后一直进行该操作直至栈为空