結果

問題 No.2006 Decrease All to Zero
ユーザー to-omerto-omer
提出日時 2022-07-01 18:37:40
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 844 ms / 4,000 ms
コード長 2,334 bytes
コンパイル時間 239 ms
コンパイル使用メモリ 82,688 KB
実行使用メモリ 86,008 KB
最終ジャッジ日時 2024-04-26 20:49:35
合計ジャッジ時間 9,707 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
52,608 KB
testcase_01 AC 37 ms
52,352 KB
testcase_02 AC 121 ms
76,248 KB
testcase_03 AC 40 ms
52,992 KB
testcase_04 AC 40 ms
52,608 KB
testcase_05 AC 40 ms
52,608 KB
testcase_06 AC 83 ms
76,416 KB
testcase_07 AC 92 ms
76,544 KB
testcase_08 AC 46 ms
54,272 KB
testcase_09 AC 46 ms
54,400 KB
testcase_10 AC 558 ms
84,284 KB
testcase_11 AC 820 ms
85,656 KB
testcase_12 AC 565 ms
83,568 KB
testcase_13 AC 571 ms
83,372 KB
testcase_14 AC 844 ms
86,008 KB
testcase_15 AC 811 ms
85,228 KB
testcase_16 AC 461 ms
80,992 KB
testcase_17 AC 629 ms
83,524 KB
testcase_18 AC 39 ms
52,272 KB
testcase_19 AC 38 ms
52,864 KB
testcase_20 AC 38 ms
52,864 KB
testcase_21 AC 38 ms
52,736 KB
testcase_22 AC 39 ms
52,608 KB
testcase_23 AC 39 ms
52,992 KB
testcase_24 AC 39 ms
52,352 KB
testcase_25 AC 39 ms
52,992 KB
testcase_26 AC 640 ms
85,548 KB
testcase_27 AC 733 ms
77,584 KB
testcase_28 AC 767 ms
78,088 KB
testcase_29 AC 196 ms
78,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

input = sys.stdin.buffer.readline
INF = 10 ** 18


class QueueAggregation:
    def __init__(self, e, f):
        self.e = e
        self.f = f
        self.front = []
        self.back = []

    def fold_all(self):
        return self.f(
            self.front[-1][0] if self.front else self.e,
            self.back[-1][0] if self.back else self.e,
        )

    def append(self, val):
        x = self.f(
            self.back[-1][0] if self.back else self.e,
            val,
        )
        self.back.append((x, val))

    def appendleft(self, val):
        x = self.f(
            val,
            self.front[-1][0] if self.front else self.e,
        )
        self.front.append((x, val))

    def pop(self):
        if not self.front:
            while self.back:
                _, val = self.back.pop()
                self.appendleft(val)
        return self.front.pop()[1]


def main():
    N = int(input())
    X = list(map(int, input().split()))
    A = list(map(int, input().split()))
    M = max(A) + 1
    DP = [[INF] * M for _ in range(3)]
    EP = [[INF] * M for _ in range(3)]
    for j in range(3):
        DP[j][A[0] * 2 - 2 >> 1] = 0
    for i in range(1, N):
        w = X[i] - X[i - 1]
        for j in range(3):
            for k in range(M):
                DP[j][k] += (k * 2 + 2 - j) * w
                EP[j][k] = INF
        for j in range(3):
            for nj in range(j, 3):
                que = QueueAggregation(INF, min)
                L, R = max(A[i] - 2 + j, int(j == 2)), min(M, A[i] + 1 - nj + j)
                for k in range(R - 1, L - 1, -1):
                    que.appendleft(DP[j][k])
                for nk in range(M):
                    nL = max(nk - A[i], A[i] - nk - 2 + j, int(j == 2))
                    nR = min(M, nk + A[i] + 1 - nj + j)
                    while L > nL:
                        L -= 1
                        que.appendleft(DP[j][L])
                    while R < nR:
                        que.append(DP[j][R])
                        R += 1
                    while L < nL:
                        que.pop()
                        L += 1
                    EP[nj][nk] = min(EP[nj][nk], que.fold_all())
        DP, EP = EP, DP
    ANS = DP[2][0]
    if ANS >= INF:
        ANS = -1
    print(ANS)


if __name__ == "__main__":
    main()
0