Create 093. 平衡二叉树.cpp

cl886699 · Sep 8, 2017 · 3ae4c5e · 3ae4c5e
1 parent 1f3744d
commit 3ae4c5e
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/093. 平衡二叉树.cpp b/093. 平衡二叉树.cpp
@@ -0,0 +1,30 @@
+/**
+ * Definition of TreeNode:
+ * class TreeNode {
+ * public:
+ *     int val;
+ *     TreeNode *left, *right;
+ *     TreeNode(int val) {
+ *         this->val = val;
+ *         this->left = this->right = NULL;
+ *     }
+ * }
+ */
+class Solution 
+{
+public:
+    bool isBalanced(TreeNode *root) 
+    {
+        if(root==NULL)
+            return true;
+        if(abs(getHeight(root->left)-getHeight(root->right))>1)
+            return false;
+        return isBalanced(root->left)&&isBalanced(root->right);
+    }
+    int getHeight(TreeNode *root)
+    {
+        if(root==NULL)
+            return 0;
+        return max(getHeight(root->left),getHeight(root->right))+1;
+    }
+};