Merge pull request #201 from jojoo15/patch-13

添加 0501.二叉搜索树中的众数 python版本
This commit is contained in:
Carl Sun
2021-05-21 09:33:53 +08:00
committed by GitHub

View File

@ -394,8 +394,39 @@ class Solution {
``` ```
Python Python
```python
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
//递归法
class Solution:
def findMode(self, root: TreeNode) -> List[int]:
if not root: return
self.pre = root
self.count = 0 //统计频率
self.countMax = 0 //最大频率
self.res = []
def findNumber(root):
if not root: return None // 第一个节点
findNumber(root.left) //左
if self.pre.val == root.val: //中: 与前一个节点数值相同
self.count += 1
else: // 与前一个节点数值不同
self.pre = root
self.count = 1
if self.count > self.countMax: // 如果计数大于最大值频率
self.countMax = self.count // 更新最大频率
self.res = [root.val] //更新res
elif self.count == self.countMax: // 如果和最大值相同放进res中
self.res.append(root.val)
findNumber(root.right) //右
return
findNumber(root)
return self.res
```
Go Go