Сложность: medium
Вы стоите в позиции 0 на бесконечной числовой прямой. В позиции target находится пункт назначения. Вы можете сделать некоторое количество ходов numMoves так, чтобы: на каждом ходу вы могли пойти либо налево, либо направо. Во время i-го хода (начиная с i == 1 до i == numMoves) вы делаете i шагов в выбранном направлении. Учитывая целое число target, верните минимальное количество ходов (т.е. минимальное numMoves), необходимое для достижения пункта назначения.
Пример:
Input: heights = [2,1,1,2,1,2,2], volume = 4, k = 3
Output: [2,2,2,3,2,2,2]
👨💻 Алгоритм:
1⃣Инициализируйте цикл для добавления объема воды.
2⃣Для каждой единицы воды: Проверьте, может ли вода двигаться влево и упасть на более низкий уровень. Если нет, проверьте, может ли вода двигаться вправо и упасть на более низкий уровень. Если нет, добавьте воду в текущую позицию.
3⃣Повторите шаг 2, пока не будет добавлен весь объем воды.
😎 Решение:
def pourWater(heights, volume, k):
for _ in range(volume):
drop_index = k
for d in (-1, 1):
i = k
while 0 <= i + d < len(heights) and heights[i + d] <= heights[i]:
if heights[i + d] < heights[i]:
drop_index = i + d
i += d
if drop_index != k:
break
heights[drop_index] += 1
return heights
Ставь 👍 и забирай 📚 Базу знаний