Найдите максимальную частоту элемента в заданном массиве
Это решение, которое я придумал, но я не уверен, является ли это лучшим возможным решением в отношении нотации Big (O)...
def solution(A):
B = [0, 0, 0, 0, 0]
for i in range (len(A)):
if A[i] == "Cardiology":
B[0] += 1
elif A[i] == "Neurology":
B[1] += 1
elif A[i] == "Orthopaedics":
B[2] += 1
elif A[i] == "Gynaecology":
B[3] += 1
elif A[i] == "Oncology":
B[4] += 1
max_patients = max(B)
return max_patients
Ответы 3
Поскольку вы знаете все возможные значения, вы можете использовать dict с названиями отделов в качестве ключей и количеством в качестве значений.
Вы можете инициализировать его как:
departments = {"Cardiology": 0, "Neurology": 0, "Orthopaedics": 0, "Gynaecology": 0, "Oncology": 0}
В качестве предложения по стилю, поскольку вы перебираете элементы списка, вам не нужно обращаться к ним по индексу, вместо этого вы можете напрямую перебирать список. Объединив это со словарем, вы можете сделать:
for dept in A:
departments[dept] += 1
max_patients = max(departments)
return max_patients
Конечно, если вы хотите немного изучить документацию, объект collections.Counter делает то же самое (но, возможно, немного быстрее).
Вы можете решить это очень легко с помощью collections.Counter. Вам не нужны словари, боковые списки или что-либо еще лишнее. Все, что вы добавляете, это еще одна вещь, которую можно сломать. Держите его как можно проще.
from collections import Counter
def solution(A):
return max(Counter(A).values())
Я только вставил это в функцию, чтобы дать вам контекст. Нет причин для того, чтобы это была функция. Оборачивая это в функцию, вы просто даете операции псевдоним. К сожалению, ваш псевдоним не дает никаких указаний на то, что вы псевдоним. Лучше просто поставить одну линию на место.
На основе предположений и возвращаемого значения:
def solution(A):
A = [A.count(A[i]) for i in set(range(len(A)))]
return max(A)
A = ["Cardiology", "Neurology", "Oncology", "Orthopaedics", "Gynaecology", "Oncology", "Oncology"]
print(solution(A))
# 3
Я не знал о collections.Counter. Спасибо!