代码随想录算法训练营第十四天|二叉树递归遍历 二叉树迭代遍历

目录

二叉树递归遍历

1.前序遍历

2.中序遍历

3.后序遍历

二叉树迭代遍历

1.前序遍历

2.中序遍历

3.后序遍历

二叉树统一迭代 

1.前序遍历

2.中序遍历

3.后序遍历

---

二叉树递归遍历

1.前序遍历

144. 二叉树的前序遍历 - 力扣（LeetCode）

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> result;
        traversal(root, result);
        return result;
    }
    void traversal(TreeNode* cur, vector<int>& vec) {
        if (cur == NULL) return;
        vec.push_back(cur->val);
        traversal(cur->left, vec);
        traversal(cur->right, vec);
    }
};

2.中序遍历

94. 二叉树的中序遍历 - 力扣（LeetCode）

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;
        traversal(root, result);
        return result;
    }
    void traversal(TreeNode* cur, vector<int>& vec) {
        if (cur == NULL) return;  
        traversal(cur->left, vec);
        vec.push_back(cur->val);
        traversal(cur->right, vec);
    }
};

3.后序遍历

145. 二叉树的后序遍历 - 力扣（LeetCode）

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> result;
        traversal(root, result);
        return result;
    }
    void traversal(TreeNode* cur, vector<int>& vec) {
        if (cur == NULL) return;
        traversal(cur->left, vec);
        traversal(cur->right, vec);
        vec.push_back(cur->val);
    }
};

---

二叉树迭代遍历

1.前序遍历

144. 二叉树的前序遍历 - 力扣（LeetCode）

思路：先处理中间结点，即根节点先入栈，右孩子入栈，再左孩子入栈。

class Solution {
public:
    vector<int> preorderTraversal(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->right) st.push(node->right);
			if (node->left)  st.push(node->left);
		}  
        return result;
    }
};

2.中序遍历

94. 二叉树的中序遍历 - 力扣（LeetCode）

思路：使用迭代法写中序遍历，需要借用指针的遍历来帮助访问节点，栈则用来处理节点上的元素。

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

3.后序遍历

145. 二叉树的后序遍历 - 力扣（LeetCode）

思路：在先序遍历的基础上做些调整，再将结果反转就是答案。

class Solution {
public:
    vector<int> postorderTraversal(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;
    }
};

---

二叉树统一迭代 

思路：将访问的结点放入栈中，把要处理的结点也放入栈中，紧接着放入一个空结点作为标记。

1.前序遍历

144. 二叉树的前序遍历 - 力扣（LeetCode）

class Solution {
public:
	vector<int> preorderTraversal(TreeNode* root) {
		vector<int> result;
		stack<TreeNode*> st;
		if (root != NULL) st.push(root);
		while (!st.empty()) {
			TreeNode* node = st.top();
			if (node != NULL) {
				st.pop();
				if (node->right) st.push(node->right);//添加右结点 
				if (node->left) st.push(node->left);  //添加左结点
				st.push(node);                        //添加中结点 
				st.push(NULL);	 
			}
			else {
				st.pop();
				node = st.top();
				st.pop();
				result.push_back(node->val);
			}
		}
		return result;
	}
}; 

2.中序遍历

94. 二叉树的中序遍历 - 力扣（LeetCode）

class Solution {
public:
	vector<int> inorderTraversal(TreeNode* root) {
		vector<int> result;
		stack<TreeNode*> st;
		if (root != NULL) st.push(root);
		while (!st.empty()) {
			TreeNode* node = st.top();
			if (node != NULL) {
				st.pop();
				if (node->right) st.push(node->right);//添加右结点 
				st.push(node);                        //添加中结点 
				st.push(NULL);
				if (node->left) st.push(node->left);  //添加左结点 
			}
			else {
				st.pop();
				node = st.top();
				st.pop();
				result.push_back(node->val);
			}
		}
		return result;
	}
}; 

3.后序遍历

145. 二叉树的后序遍历 - 力扣（LeetCode）

class Solution {
public:
	vector<int> postorderTraversal(TreeNode* root) {
		vector<int> result;
		stack<TreeNode*> st;
		if (root != NULL) st.push(root);
		while (!st.empty()) {
			TreeNode* node = st.top();
			if (node != NULL) {
				st.pop();
				st.push(node);                        //添加中结点 
				st.push(NULL);	 
				if (node->right) st.push(node->right);//添加右结点 
				if (node->left) st.push(node->left);  //添加左结点
			}
			else {
				st.pop();
				node = st.top();
				st.pop();
				result.push_back(node->val);
			}
		}
		return result;
	}
};