题意: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;
}