Leetcode-- 合法二叉搜索樹

實現一個函數,檢查一棵二叉樹是否爲二叉搜索樹。

示例 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);
    }
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章