結果

問題 No.1117 数列分割
ユーザー vwxyzvwxyz
提出日時 2024-04-14 12:36:14
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,636 ms / 3,000 ms
コード長 937 bytes
コンパイル時間 246 ms
コンパイル使用メモリ 82,956 KB
実行使用メモリ 111,332 KB
最終ジャッジ日時 2024-10-03 10:15:50
合計ジャッジ時間 17,381 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque

def Slide_Min(lst,width):
    queue=deque([])
    for i,x in enumerate(lst[:width-1]):
        while queue and x<=lst[queue[-1]]:
            queue.pop()
        queue.append(i)
    slide_min=[]
    for i,x in enumerate(lst[width-1:],width-1):
        while queue and x<=lst[queue[-1]]:
            queue.pop()
        queue.append(i)
        slide_min.append(lst[queue[0]])
        if queue[0]==i-width+1:
            queue.popleft()
    return slide_min

N,K,M=map(int,input().split())
A=[0]+list(map(int,input().split()))
for i in range(1,N+1):
    A[i]+=A[i-1]
inf=1<<60
dp=[-inf]*(N+1)
dp[0]=0
for k in range(1,K+1):
    prev=dp
    dp=[-inf]*(N+1)
    for i,x in enumerate(Slide_Min([inf]*M+[-A[i]-prev[i] for i in range(N)],M)):
        dp[i]=max(dp[i],-A[i]-x)
    for i,x in enumerate(Slide_Min([inf]*M+[A[i]-prev[i] for i in range(N)],M)):
        dp[i]=max(dp[i],A[i]-x)
    ans=dp[N]
print(ans)
0