- 題目:判斷一顆二叉樹是否為BST。
- 思路:其實這個問題可以有多個解決方法。
- 方法一:遞歸解決。根據BST的特性。左邊的小於根節點的值,右邊的大於根節點的值。並且對於每一棵子樹都是如此。所以我們可以直接遞歸的對左右子樹的值與根節點的值進行比較。左子樹的值小於當前根節點的值,將當前根節點的值作為最大值傳入左子樹,左子樹的值都小於他,遞歸處理;右子樹的值都大於根節點的值,將根節點的值作為最小值傳入右子樹,右子樹的值都大於他。
- 代碼:
/** * Definition for binary tree * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */ class Solution { public: bool isValidBST(TreeNode *root) { return isValidBST(root, INT_MIN, INT_MAX); } bool isValidBST(TreeNode *root, int low, int high){ if (root == NULL ) return true; if (low < root->val && root->val < high) return (isValidBST(root->left, low, root->val) && isValidBST(root->right, root->val, high)); else return false; } };
- 方法二:因為BST特性,所以我們可以利用遍歷方法對他進行解決。對樹進行中序遍歷,將結果存儲在vector中,如果容器中的值是遞增排序的,那么它就是BST,否則就不是。
- 代碼:
/** * Definition for binary tree * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */ class Solution { public: bool isValidBST(TreeNode *root) { vector<int> res; isValidBST(root, res); int len = res.size(); bool flag = true; for (int i=0; i<len-1; i++){ if (res[i] >= res[i+1]){ flag = false; break; } } return flag; } void isValidBST(TreeNode *root, vector<int> &res){ if (root == NULL) return; isValidBST(root->left, res); res.push_back(root->val); isValidBST(root->right, res); } };