二叉树前序遍历
- 我的天,我发现,前序遍历
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> *result=new vector<int>;
if(!root)
return *result;
stack<struct TreeNode*> st;
st.push(root);
while (!st.empty())
{
struct TreeNode* temp = st.top();
st.pop();
(*result).push_back(temp->val);
if (temp->right)
st.push(temp->right);
if (temp->left)
st.push(temp->left);
}
return *result;
}
};
- 其实可以把result申明为局部的数组
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> result;
if(!root)
return result;
stack<struct TreeNode*> st;
st.push(root);
while (!st.empty())
{
struct TreeNode* temp = st.top();
st.pop();
result.push_back(temp->val);
if (temp->right)
st.push(temp->right);
if (temp->left)
st.push(temp->left);
}
return result;
}
};
二叉树中遍历有点复杂
- 怪不得我没写出来呢
二叉树前序、中序、后序遍历—迭代方法
- 前序遍历很简单,
- 先把当前节点的值保存到ret中,
- 再把它的右孩子、左孩子分别压入stack中。
- 先右后左。
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> ret;
stack<TreeNode*> s;
s.push(root);
while(!s.empty()){
TreeNode* cur = s.top();
s.pop();
if(cur != NULL){
ret.push_back(cur->val);
s.push(cur->right);
s.push(cur->left);
}
}
return ret;
}
};
- 中序遍历,稍微复杂点:
- 根据中序遍历的顺序,对任一结点,优先访问其左孩子,左孩子又可以看做一根结点,然后继续访问其左孩子结点,直到遇到左孩子结点为空的结点才访问,然后按相同的规则访问其 右子树。
- 如下:
- 对任一结点p
- 1.若其左孩子不为空,
- 则将p入栈并将p的左孩子置为当前p,然后对当前结点p相同的处理;
- 2.若其左孩子为空,则取栈顶元素并出栈,
- 访问该栈顶结点,
- 然后将栈顶结点的右孩子置为当前p
- 3.p为null且栈为空则结束