src/$tests/data_structure/CHT.test.py

View this file on GitHub
Last update: 2023-07-23 08:45:17+09:00
Problem: https://yukicoder.me/problems/no/409
Link: View error logs on GitHub Actions

Depends on

src/data_structure/ConvexHullTrick.py

Code

# verification-helper: PROBLEM https://yukicoder.me/problems/no/409
from pathlib import Path
import sys

sys.path.append(str(Path(__file__).resolve().parent.parent.parent.parent))
from src.data_structure.ConvexHullTrick import ConvexHullTrick


n, a, b, w = map(int, input().split())
D = list(map(int, input().split()))
dp = [0] * (n + 1)
dp[0] = w


ch = ConvexHullTrick()
for i, d in enumerate(D):
    ch.add_line(-2 * b * i, 2 * dp[i] + 2 * a * i + b * i * i + b * i)
    x = i + 1
    mi = ch.get(x)
    tmp = mi + 2 * a + 2 * d - 2 * a * x + b * x * x - b * x
    dp[x] = tmp // 2

ans = 1 << 60
for i in range(n + 1):
    d = n - i
    ans = min(ans, dp[i] - a * d + b * d * (d + 1) // 2)
print(ans)

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.11.4/x64/lib/python3.11/site-packages/onlinejudge_verify/documentation/build.py", line 81, in _render_source_code_stat
    bundled_code = language.bundle(
                   ^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.11.4/x64/lib/python3.11/site-packages/onlinejudge_verify/languages/python.py", line 108, in bundle
    raise NotImplementedError
NotImplementedError