結果

問題 No.489 株に挑戦
ユーザー nanae
提出日時 2017-02-24 23:38:41
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,539 bytes
コンパイル時間 330 ms
コンパイル使用メモリ 12,416 KB
実行使用メモリ 26,752 KB
最終ジャッジ日時 2025-01-03 00:39:57
合計ジャッジ時間 4,481 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 6 WA * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from itertools import accumulate

def debug(x, table):
    for name, val in table.items():
        if x is val:
            print('DEBUG:{} -> {}'.format(name, val), file=sys.stderr)
            return None

def solve():
    N, D, K = map(int, input().split())
    X = [int(sys.stdin.readline().rstrip()) for i in range(N)]
    Xac = list(accumulate(X))
    mims = []
    maxs = []

    for i in range(N):
        if i == 0:
            if X[0] < X[1]:
                mims.append((i, X[i]))
        elif i == N - 1:
            if X[-1] > X[-2]:
                maxs.append((i, X[i]))
        else:
            if X[i - 1] < X[i] and X[i] >= X[i + 1]:
                maxs.append((i, X[i]))
            elif X[i - 1] >= X[i] and X[i] < X[i + 1]:
                mims.append((i, X[i]))

    max_dif = 0
    j, k = None, None

    for p1, p2 in zip(mims, maxs):
        if p2[0] - p1[0] <= D:
            if p2[1] - p1[1] > max_dif:
                max_dif = p2[1] - p1[1]
                j, k = p1[0], p2[0]
        else:
            j1, k1, dif = find_maxdif(Xac, p1[0], p2[0], D)
            if dif > max_dif:
                j, k = j1, k1
                max_dif = dif

    if max_dif <= 0:
        print(0)
    else:
        print(max_dif * K)
        print(j, k)

def find_maxdif(Xac, l, r, D):
    res = 0
    j, k = 0, 0

    for i in range(l, r + 1 - D):
        if Xac[i + D] - Xac[i] > res:
            res = Xac[i + D] - Xac[i]
            j, k = i, i + D

    return j, k, res

if __name__ == '__main__':
    solve()
0