結果
問題 |
No.409 ダイエット
|
ユーザー |
|
提出日時 | 2025-06-29 01:19:21 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 476 ms / 2,000 ms |
コード長 | 847 bytes |
コンパイル時間 | 259 ms |
コンパイル使用メモリ | 82,184 KB |
実行使用メモリ | 124,320 KB |
最終ジャッジ日時 | 2025-06-29 01:19:36 |
合計ジャッジ時間 | 14,604 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 92 |
ソースコード
n, a, b, w = map(int, input().split()) D = [0]+list(map(int, input().split()))+[0] if b==0: print(w-n*a) exit() dp = [1<<60]*-~n dp[0] = w calcx = lambda j: dp[j]+(j+1)*a+b*(j**2+j)//2 calc = lambda i, j: (calcx(i)-calcx(j))/(b*(i-j)) calcmin = lambda i, j: calcx(j)-b*i*j calcall = lambda i, j: D[i]-i*a+b*(i**2-i)//2+calcmin(i, j) from collections import deque stack = deque([0]) ans = 0 num = 0 length = 1 for i in range(1, n+2): d = D[i] while length>=2 and calcmin(i, stack[0])>=calcmin(i, stack[1]): stack.popleft() length -= 1 if i==n+1: # print(dp) print(dp[stack[0]]+(n-stack[0])*(n-stack[0]+1)//2*b-a*(n-stack[0])) exit() dp[i] = calcall(i, stack[0]) while length>=2 and calc(stack[-1], stack[-2])>=calc(stack[-1], i): stack.pop() length -= 1 stack.append(i) length += 1 # print(stack)