LC 94,98,230 (栈实现的中序遍历三连)

题意:LC94 中序遍历BST,LC98 Validate BST,LC230 Kth Smallest element in BST。

LC 94 中序遍历BST

迭代的方法中序遍历BST,依赖栈实现,当前节点不为空时一直将左孩子压栈,当节点为空时说明左孩子为空,此时左子树遍历完毕,元素出栈得到父节点值,然后将右孩子压栈,循环往复。注意大循环的判断条件为while( root || ! st.empty() )

vector<int> inorderTraversal(TreeNode* root){
        vector<int> res;
        stack<TreeNode*> st;
        while (root||!st.empty()) {
            while (root) {
                st.push(root);
                root = root->left;
            }
            root = st.top();
            st.pop();
            res.push_back(root->val);
            root = root->right;
        }
        return res;
    }
LC 98 判断BST是否有效
BST有一个非常重要的性质那就是中序遍历一个BST得到的数列时有序递增的。利用此规律,对于任何一个二叉树,通过中序遍历二叉树判断所得数列是否有序即可。代码实现的时候可以完全照搬LC94的代码,最后多加一步判断res数组有序。也可以在迭代的过程中增加一个变量 pre 表示上一个遍历到的节点,代码如下:
bool isValidBST(TreeNode* root) {
    if (root == NULL) return true;
    stack<TreeNode*> st;
    TreeNode* pre = NULL;
    while (root != NULL || !st.empty()) {
        while (root != NULL) {
            st.push(root);
            root = root->left;
        }
        root = st.top();
        st.pop();
        if(pre != NULL && root->val <= pre->val) return false;
        pre = root;
        root = root->right;
    }
    return true;
}


LC 230 Kth smallest element in BST

根据上面提到的规律,中序遍历BST并在每次出栈时对 k -= 1记录当前元素是第几小,直到k == 0得到所求的第k小元素。

int kthSmallest(TreeNode* root, int k) {
    stack<TreeNode*> st;
    while (root || !st.empty()) {
        while (root) {
            st.push(root);
            root = root->left;
        }
        root = st.top();
        st.pop();
        if (--k == 0) {
            break;
        }
        root = root->right;
    }
    return root->val;
}

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值