有关于二叉树的题目,特别是给你一个root(根节点)让你做题 的,大都离不开这两种遍历:1.深度遍历;2.层序遍历。
做题做着做着就感觉都是一个套路,遍历过程中加一些条件得出对应内容就好了。
力扣题目:
一.深度遍历
做题时基本用的是前序遍历,有时候用到后序遍历,几乎没有遇到中序。
所以把前序弄清楚,做题也就能轻松AC了。
我比较喜欢用迭代法,递归如果不是必须要用,我是不用的。
深度遍历用栈来模拟二叉树,前序遍历代码如下:
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;
}
};
二.层序遍历
层序遍历用队列进行模拟,层序遍历更容易理解,有时某些题两种遍历都可以用我会优先选择层序遍历。
代码如下:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> que;
if (root != NULL) que.push(root);
vector<vector<int>> result;
while (!que.empty()) {
int size = que.size();
vector<int> vec;
// 这里一定要使用固定大小size,不要使用que.size(),因为que.size是不断变化的
for (int i = 0; i < size; i++) {
TreeNode* node = que.front();
que.pop();
vec.push_back(node->val);
if (node->left) que.push(node->left);
if (node->right) que.push(node->right);
}
result.push_back(vec);
}
return result;
}
};
暂无评论