forked from shichao-an/leetcode-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsolution.py
More file actions
23 lines (22 loc) · 760 Bytes
/
solution.py
File metadata and controls
23 lines (22 loc) · 760 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# Definition for a binary tree node
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
# @param root, a tree node
# @return an integer
def minDepth(self, root):
if root is None:
return 0
if root.left is None and root.right is None:
return 1
elif root.left is None and root.right is not None:
return self.minDepth(root.right) + 1
elif root.left is not None and root.right is None:
return self.minDepth(root.left) + 1
else:
left_min = self.minDepth(root.left)
right_min = self.minDepth(root.right)
return min(left_min, right_min) + 1