Это была одна из проблем в текущем 30-дневном вызове лецкода. Опубликовано 7 апреля 2020 года.
Учитывая целочисленное массив ARR, Count Element X такой, что X + 1 также в ARR.
Если в массиве есть дубликаты, подсчитайте их отдельно.
Пример 1:
Input: arr = [1,2,3] Output: 2 Explanation: 1 and 2 are counted cause 2 and 3 are in arr.
Пример 2:
Input: arr = [1,1,3,3,5,5,7,7] Output: 0 Explanation: No numbers are counted, cause there's no 2, 4, 6, or 8 in arr.
Пример 3:
Input: arr = [1,3,2,3,5,0] Output: 3 Explanation: 0, 1 and 2 are counted cause 1, 2 and 3 are in arr.
Пример 4:
Input: arr = [1,1,2,2] Output: 2 Explanation: Two 1s are counted cause 2 is in arr.
Подход решения
- Как оригинальный список содержит дубликаты элементов. Преобразуйте список, чтобы установить, чтобы получить уникальные элементы.
- Имейте переход на множество и проверьте, существует ли элемент + 1 в исходном списке.
- Если это существует. Найдите счет элемента в исходном списке.
- Продолжайте добавлять счет в новый список.
- Сумма всех средств. Возвращает необходимый выход.
Кодовый фрагмент
class Solution: def countElements(self, arr: List[int]) -> int: unique_elements = set(arr) L = [] for element in unique_elements: if element + 1 in arr: item_count = arr.count(element) L.append(item_count) return sum(L)
Добро пожаловать на более оптимизированные решения.
Оригинал: “https://dev.to/mridubhatnagar/given-an-integer-array-arr-count-element-x-such-that-x-1-is-also-in-arr-332k”