|
104 | 104 | "source": [ |
105 | 105 | "class BstBalance(Bst):\n", |
106 | 106 | "\n", |
107 | | - " def _check_height(self, node):\n", |
| 107 | + " def _check_balance(self, node):\n", |
108 | 108 | " if node is None:\n", |
109 | 109 | " return 0\n", |
110 | | - " left_height = self._check_height(node.left)\n", |
| 110 | + " left_height = self._check_balance(node.left)\n", |
111 | 111 | " if left_height == -1:\n", |
112 | 112 | " return -1\n", |
113 | | - " right_height = self._check_height(node.right)\n", |
| 113 | + " right_height = self._check_balance(node.right)\n", |
114 | 114 | " if right_height == -1:\n", |
115 | 115 | " return -1\n", |
116 | | - " diff = abs(left_height-right_height)\n", |
| 116 | + " diff = abs(left_height - right_height)\n", |
117 | 117 | " if diff > 1:\n", |
118 | 118 | " return -1\n", |
119 | 119 | " return 1 + max(left_height, right_height)\n", |
120 | 120 | "\n", |
121 | 121 | " def check_balance(self):\n", |
122 | 122 | " if self.root is None:\n", |
123 | 123 | " raise TypeError('root cannot be None')\n", |
124 | | - " height = self._check_height(self.root)\n", |
| 124 | + " height = self._check_balance(self.root)\n", |
125 | 125 | " return height != -1" |
126 | 126 | ] |
127 | 127 | }, |
|
235 | 235 | "name": "python", |
236 | 236 | "nbconvert_exporter": "python", |
237 | 237 | "pygments_lexer": "ipython3", |
238 | | - "version": "3.5.0" |
| 238 | + "version": "3.4.3" |
239 | 239 | } |
240 | 240 | }, |
241 | 241 | "nbformat": 4, |
|
0 commit comments