實現一個函數,檢查一棵二叉樹是否爲二叉搜索樹。
示例 1:
輸入:
2
/ \
1 3
輸出: true
示例 2:
輸入:
5
/ \
1 4
/ \
3 6
輸出: false
解釋: 輸入爲: [5,1,4,null,null,3,6]。
根節點的值爲 5 ,但是其右子節點值爲 4 。
思路
利用搜索樹的中序遍歷爲升序的特性來判斷樹的合法性;
如果中序遍歷的結果爲升序,說明是一顆二叉搜索樹;否則不是
/**
* 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:
bool check(TreeNode* root, TreeNode*& preT){
// 遍歷到了空節點
if(!root) return true;
bool left= check(root->left, preT);
// 中序遍歷的前一個節點的值如果大於等於當前節點的值,說明搜索樹不合法
if(preT&& preT->val>= root->val)
return false;
preT= root;
bool right= check(root->right, preT);
return left&& right;
}
bool isValidBST(TreeNode* root) {
// preT保存程序遍歷的前一個節點
TreeNode* preT= NULL;
return check(root, preT);
}
};