結果

問題 No.1117 数列分割
ユーザー first_vilfirst_vil
提出日時 2020-07-04 17:09:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,981 ms / 3,000 ms
コード長 973 bytes
コンパイル時間 234 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 161,240 KB
最終ジャッジ日時 2024-05-07 11:16:42
合計ジャッジ時間 18,003 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
53,504 KB
testcase_01 AC 47 ms
53,888 KB
testcase_02 AC 46 ms
53,888 KB
testcase_03 AC 247 ms
78,996 KB
testcase_04 AC 158 ms
77,976 KB
testcase_05 AC 43 ms
53,632 KB
testcase_06 AC 112 ms
76,544 KB
testcase_07 AC 103 ms
76,372 KB
testcase_08 AC 171 ms
78,084 KB
testcase_09 AC 159 ms
77,904 KB
testcase_10 AC 190 ms
78,548 KB
testcase_11 AC 401 ms
79,896 KB
testcase_12 AC 476 ms
81,120 KB
testcase_13 AC 570 ms
83,516 KB
testcase_14 AC 625 ms
82,540 KB
testcase_15 AC 767 ms
84,896 KB
testcase_16 AC 803 ms
84,496 KB
testcase_17 AC 197 ms
78,400 KB
testcase_18 AC 1,981 ms
161,240 KB
testcase_19 AC 1,423 ms
99,328 KB
testcase_20 AC 719 ms
83,656 KB
testcase_21 AC 1,319 ms
94,840 KB
testcase_22 AC 1,332 ms
94,776 KB
testcase_23 AC 1,292 ms
94,656 KB
testcase_24 AC 1,226 ms
93,824 KB
testcase_25 AC 1,214 ms
93,696 KB
testcase_26 AC 47 ms
54,016 KB
testcase_27 AC 222 ms
79,420 KB
testcase_28 AC 204 ms
78,372 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque
n,k,m=map(int,input().split())
a=list(map(int,input().split()))+[0]

for i in range(n-1,-1,-1):
    a[i]+=a[i+1]

plus=[deque([]) for j in range(k)]
plus[0].append((-1,a[0]))
minus=[deque([]) for j in range(k)]
minus[0].append((-1,-a[0]))

for i in range(n):
    for j in range(k-1,-1,-1):
        res=0
        while plus[j] and plus[j][0][0]<i-m:
            plus[j].popleft()
        while minus[j] and minus[j][0][0]<i-m:
            minus[j].popleft()
        if plus[j]:
            res=max(res,plus[j][0][1]-a[i+1])
        if minus[j]:
            res=max(res,minus[j][0][1]+a[i+1])
        if j+1<k:
            while plus[j+1] and plus[j+1][-1][1]<=res+a[i+1]:
                plus[j+1].pop()
            while minus[j+1] and minus[j+1][-1][1]<=res-a[i+1]:
                minus[j+1].pop()
            plus[j+1].append((i,res+a[i+1]))
            minus[j+1].append((i,res-a[i+1]))
        if i==n-1 and j==k-1:
            print(res)
0