瀏覽代碼

angelic

master
Roderic Day 3 年之前
父節點
當前提交
1f608f8fbc
共有 1 個文件被更改,包括 12 次插入0 次删除
  1. +12
    -0
      y2021/p07.py

+ 12
- 0
y2021/p07.py 查看文件

ns = [int(n) for n in text.split(',')]
lo, *_, hi = sorted(ns)

def mono_min(g):
y = next(g)
for x in g:
if x > y: return y
y = x

ans1 = mono_min(sum(abs(n - m) for m in ns) for n in range(lo, hi))
range_sum = lambda n: int(n * (n + 1) / 2)
ans2 = mono_min(sum(range_sum(abs(n - m)) for m in ns) for n in range(lo, hi))

Loading…
取消
儲存