华为OD机试 Python【小区数量】

题目

幼儿园组织活动，老师布置了一个任务：
每个小朋友去了解与自己同一个小区的小朋友还有几个。
我们将这些数量汇总到数组 garden 中。
请根据这些小朋友给出的信息，计算班级小朋友至少来自几个小区？

输入描述
输入：2 2 3

输出描述
输出：7

解题思路

基于统计每个小区的人数，找到最大的小区人数。然后将所有小区的人数总和与最大的小区人数进行比较，取两者中较大的值作为结果，确保每个小朋友至少来自一个不同的小区。

代码

# -*- coding: utf-8 -*-

def count_distinct_areas(garden):
    total = 0
    max_area = 0
    for num <