二叉树非递归(迭代)遍历

保存父亲结点到自建栈内,也就是一路沿着左边压栈,回头了就弹出父结点,再走别的路。(树之所以要用栈,是因为下去了就没法上去)

//前序遍历
vector<int> preorderTraversal(TreeNode* root) {
    vector<int> TraArr;
    stack<TreeNode*> NodeStk;               
    TreeNode* p=root;                       //p指针指向当前根结点
    TreeNode* q=NULL;                       //q指针指向栈顶元素,即父结点

    while(p || !NodeStk.empty()){
        if(p){
            TraArr.push_back(p->val);       //前序遍历,先把根结点的值放到遍历序列里
            NodeStk.push(p);                //父结点压栈
            q=p;                            //q指针指向栈顶元素
            p=p->left;                      //p指针指向左孩子
        }
        else{                               //若p指针为空,去看看右孩子
            p=q->right;
            NodeStk.pop();
            if(!NodeStk.empty()){
                q=NodeStk.top();            //q指针重新指向栈顶元素,即上一级父结点
            }else{
                q=NULL;
            }
        }
    }

    return TraArr;
}



//中序遍历
vector<int> inorderTraversal(TreeNode* root) {
    vector<int> TraArr;
    stack<TreeNode*> NodeStk;
    TreeNode* p=root;


    while(p || !NodeStk.empty()){
        if(p){
            NodeStk.push(p);
            p=p->left;
        }
        else{
            //由于是中序遍历,父结点值应当在左孩子之后再插入
            TraArr.push_back(NodeStk.top()->val);
            p=NodeStk.top()->right;
            NodeStk.pop();
        }
    }
        
    return TraArr;
}




//后序遍历
vector<int> postorderTraversal(TreeNode* root) {
    vector<int> TraArr;
    stack<TreeNode*> NodeStk;
    TreeNode* p=root;                          
    TreeNode* temp=NULL;                //临时存储访问过的右孩子
        
    while(p || !NodeStk.empty()){
        if(p){                          
            NodeStk.push(p);
            p=p->left;
        }else{                                  
            TreeNode* node=NodeStk.top();           
            if(node->right && node->right!=temp){    //右孩子不为空、且右孩子没被访问过的情况
                p=node->right;                       //去访问右孩子   

            }else{                                  //父结点没有右孩子、或者右孩子被访问过了的情况
                TraArr.push_back(node->val);
                NodeStk.pop();
                temp=node;                          //父结点的值插入遍历序列,并标为“访问过”状态
            }
        }
    }

    return TraArr;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值