結果

問題 No.409 ダイエット
ユーザー 0w1
提出日時 2017-08-16 02:48:30
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 963 bytes
コンパイル時間 209 ms
コンパイル使用メモリ 82,088 KB
実行使用メモリ 119,512 KB
最終ジャッジ日時 2024-10-13 11:16:54
合計ジャッジ時間 14,973 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 1 RE * 91
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque

N, A, B, W = map( int, input().split() )
D = list( map( int, input().split() ) )
dp = [ 0 for i in range( N + 1 ) ]

def get_cost( x, rb ):
  d = rb - x - 1
  return dp[ x ] + d * ( d + 1 ) // 2 * B - d * A + D[ rb - 1 ]

dq = deque()
dp[ 0 ] = W
dq.append( [ 0, 1, N ] )
for i in range( 1, N + 1 ):
  while dq[ 0 ][ 2 ] < i:
    dq.popleft()
  dp[ i ] = get_cost( dq[ 0 ][ 0 ], i )
  while len( dq ):
    x, lb, rb = dq[ len( dq ) - 1 ]
    if get_cost( x, lb ) >= get_cost( i, lb ):
      dq.pop()
      if len( dq ):
        dq[ len( dq ) - 1 ][ 2 ] = N
    else:
      break
  x, lb, rb = dq[ len( dq ) - 1 ][ 0 ], i + 1, N + 1
  while rb - lb > 1:
    mid = lb + rb >> 1
    if get_cost( x, mid ) >= get_cost( i, mid ):
      rb = mid 
    else:
      lb = mid
  dq[ len( dq ) - 1 ][ 2 ] = rb - 1
  dq.append( [ i, rb, N ] )

print( min( dp[ i ] + ( N - i ) * ( N - i + 1 ) // 2 * B - ( N - i ) * A for i in range( N + 1 ) ) )
0