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;
}

阅读更多
个人分类: LeetCode
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭
关闭