#871486 (PyPy3) No.409 ダイエット

提出ソース

結果

問題	No.409 ダイエット
ユーザー	ayaoni
提出日時	2023-05-24 00:42:21
言語	PyPy3 (7.3.15)
結果	AC
実行時間	194 ms / 2,000 ms
コード長	2,338 bytes
コンパイル時間	303 ms
コンパイル使用メモリ	82,224 KB
実行使用メモリ	126,052 KB
最終ジャッジ日時	2024-06-02 11:46:46
合計ジャッジ時間	11,536 ms
ジャッジサーバーID （参考情報）	judge5 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	35 ms 54,668 KB
testcase_01	AC	38 ms 54,952 KB
testcase_02	AC	36 ms 55,420 KB
testcase_03	AC	34 ms 55,344 KB
testcase_04	AC	35 ms 54,912 KB
testcase_05	AC	34 ms 55,500 KB
testcase_06	AC	35 ms 55,836 KB
testcase_07	AC	35 ms 54,732 KB
testcase_08	AC	35 ms 54,456 KB
testcase_09	AC	35 ms 54,324 KB
testcase_10	AC	35 ms 54,684 KB
testcase_11	AC	35 ms 55,504 KB
testcase_12	AC	34 ms 55,680 KB
testcase_13	AC	34 ms 55,080 KB
testcase_14	AC	35 ms 54,716 KB
testcase_15	AC	35 ms 54,648 KB
testcase_16	AC	36 ms 55,040 KB
testcase_17	AC	34 ms 54,792 KB
testcase_18	AC	34 ms 55,684 KB
testcase_19	AC	34 ms 55,040 KB
testcase_20	AC	38 ms 55,632 KB
testcase_21	AC	37 ms 55,852 KB
testcase_22	AC	36 ms 54,584 KB
testcase_23	AC	34 ms 54,204 KB
testcase_24	AC	34 ms 54,520 KB
testcase_25	AC	34 ms 54,484 KB
testcase_26	AC	33 ms 55,308 KB
testcase_27	AC	34 ms 54,524 KB
testcase_28	AC	34 ms 54,400 KB
testcase_29	AC	34 ms 54,756 KB
testcase_30	AC	34 ms 56,020 KB
testcase_31	AC	34 ms 54,384 KB
testcase_32	AC	33 ms 54,048 KB
testcase_33	AC	34 ms 54,760 KB
testcase_34	AC	33 ms 54,188 KB
testcase_35	AC	56 ms 72,324 KB
testcase_36	AC	67 ms 77,280 KB
testcase_37	AC	59 ms 74,192 KB
testcase_38	AC	55 ms 73,060 KB
testcase_39	AC	61 ms 73,484 KB
testcase_40	AC	61 ms 74,848 KB
testcase_41	AC	58 ms 74,392 KB
testcase_42	AC	54 ms 70,608 KB
testcase_43	AC	56 ms 71,676 KB
testcase_44	AC	64 ms 75,208 KB
testcase_45	AC	59 ms 72,056 KB
testcase_46	AC	59 ms 73,648 KB
testcase_47	AC	56 ms 71,000 KB
testcase_48	AC	60 ms 72,356 KB
testcase_49	AC	55 ms 70,956 KB
testcase_50	AC	56 ms 71,876 KB
testcase_51	AC	63 ms 74,004 KB
testcase_52	AC	57 ms 69,224 KB
testcase_53	AC	63 ms 75,188 KB
testcase_54	AC	61 ms 73,844 KB
testcase_55	AC	129 ms 100,964 KB
testcase_56	AC	130 ms 126,052 KB
testcase_57	AC	157 ms 118,204 KB
testcase_58	AC	107 ms 97,180 KB
testcase_59	AC	161 ms 107,208 KB
testcase_60	AC	113 ms 95,180 KB
testcase_61	AC	151 ms 120,084 KB
testcase_62	AC	162 ms 125,560 KB
testcase_63	AC	171 ms 120,232 KB
testcase_64	AC	141 ms 99,920 KB
testcase_65	AC	175 ms 120,476 KB
testcase_66	AC	186 ms 125,908 KB
testcase_67	AC	160 ms 115,100 KB
testcase_68	AC	161 ms 107,132 KB
testcase_69	AC	183 ms 119,808 KB
testcase_70	AC	167 ms 109,768 KB
testcase_71	AC	129 ms 99,400 KB
testcase_72	AC	171 ms 120,536 KB
testcase_73	AC	193 ms 113,336 KB
testcase_74	AC	125 ms 105,528 KB
testcase_75	AC	173 ms 113,004 KB
testcase_76	AC	142 ms 111,792 KB
testcase_77	AC	134 ms 99,484 KB
testcase_78	AC	129 ms 102,800 KB
testcase_79	AC	124 ms 97,120 KB
testcase_80	AC	162 ms 115,356 KB
testcase_81	AC	194 ms 113,684 KB
testcase_82	AC	143 ms 112,132 KB
testcase_83	AC	151 ms 112,988 KB
testcase_84	AC	152 ms 107,976 KB
testcase_85	AC	76 ms 77,696 KB
testcase_86	AC	142 ms 105,112 KB
testcase_87	AC	154 ms 109,784 KB
testcase_88	AC	117 ms 93,852 KB
testcase_89	AC	143 ms 112,520 KB
testcase_90	AC	116 ms 91,680 KB
testcase_91	AC	154 ms 116,456 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

import sys
from collections import deque
sys.setrecursionlimit(10**7)
def I(): return int(sys.stdin.readline().rstrip())
def MI(): return map(int,sys.stdin.readline().rstrip().split())
def LI(): return list(map(int,sys.stdin.readline().rstrip().split()))
def LI2(): return list(map(int,sys.stdin.readline().rstrip()))
def S(): return sys.stdin.readline().rstrip()
def LS(): return list(sys.stdin.readline().rstrip().split())
def LS2(): return list(sys.stdin.readline().rstrip())


class ConvexHullTrick:
    """
    「1次関数の追加」「あるxにおける1次関数群の最小値の取得」を高速に計算できる。
    傾きa_iは単調、xは昇順と仮定している。
    """
    def __init__(self):
        self.lines = deque()

    # y = a_1 * x + b1 がminを考えるとき無駄であるか否か(a_0 > a_1 > a_2)
    def check(self, a0, b0, a1, b1, a2, b2):
        return (b2 - b1) * (a0 - a1) <= (b1 - b0) * (a1 - a2)

    # y = a * x + b を追加(a_iが降順のとき)
    def add(self, a, b):
        while len(self.lines) >= 2:
            a0, b0 = self.lines[-2]
            a1, b1 = self.lines[-1]
            if self.check(a0, b0, a1, b1, a, b):
                self.lines.pop()
            else:
                break
        self.lines.append((a, b))

    # y = a * x + b を追加(a_iが昇順のとき)
    def add2(self, a, b):
        while len(self.lines) >= 2:
            a1, b1 = self.lines[0]
            a2, b2 = self.lines[1]
            if self.check(a, b, a1, b1, a2, b2):
                self.lines.popleft()
            else:
                break
        self.lines.appendleft((a, b))

    # x における1次関数群の最小値を求める
    def get_min(self, x):
        a, b = self.lines[0]
        res = a * x + b
        while len(self.lines) >= 2:
            a1, b1 = self.lines[1]
            y1 = a1 * x + b1
            if y1 < res:
                res = y1
                self.lines.popleft()
            else:
                break
        return res


N, A, B, W = MI()
D = [0] + LI() + [0]
inf = 10 ** 18

dp = [inf] * (N + 2)
dp[0] = 0
CHT = ConvexHullTrick()
for i in range(1, N + 2):
    CHT.add(-B * (i - 1), dp[i - 1] + i * (i - 1) // 2 * B + A * (i - 1))
    dp[i] = D[i] - A * (i - 1) + i * (i - 1) // 2 * B + CHT.get_min(i)

ans = W + dp[-1]
print(ans)

yukicoder

結果

テストケース

ソースコード