結果

問題 No.1117 数列分割
コンテスト
ユーザー vwxyz
提出日時 2024-04-14 12:36:04
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 937 bytes
コンパイル時間 164 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 18,340 KB
最終ジャッジ日時 2024-10-03 10:15:23
合計ジャッジ時間 10,679 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8 TLE * 1 -- * 17
権限があれば一括ダウンロードができます

ソースコード

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